Skip to content

Commit 69a68ba

Browse files
authored
Add nvidia-nccl-cu12 dependency to test_cuda.yml
Signed-off-by: Jinzhe Zeng <[email protected]>
1 parent 834f97e commit 69a68ba

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

.github/workflows/test_cuda.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -50,7 +50,7 @@ jobs:
5050
pip install --find-links "https://www.paddlepaddle.org.cn/packages/nightly/cu126/paddlepaddle-gpu/" --index-url https://pypi.org/simple --trusted-host www.paddlepaddle.org.cn --trusted-host paddlepaddle.org.cn "paddlepaddle-gpu==3.4.0.dev20260310"
5151
source/install/uv_with_retry.sh pip install --system -v -e .[gpu,test,lmp,cu12,torch,jax] mpi4py --reinstall-package deepmd-kit
5252
# See https://github.com/jax-ml/jax/issues/29042
53-
source/install/uv_with_retry.sh pip install --system -U 'nvidia-cublas-cu12>=12.9.0.13'
53+
source/install/uv_with_retry.sh pip install --system -U 'nvidia-cublas-cu12>=12.9.0.13' 'nvidia-nccl-cu12>=2.28.9'
5454
env:
5555
DP_VARIANT: cuda
5656
DP_ENABLE_NATIVE_OPTIMIZATION: 1

0 commit comments

Comments
 (0)