We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 55a1d37 commit 67f15f3Copy full SHA for 67f15f3
1 file changed
.github/workflows/test_cuda.yml
@@ -50,7 +50,7 @@ jobs:
50
pip install --find-links "https://www.paddlepaddle.org.cn/packages/nightly/cu126/paddlepaddle-gpu/" --index-url https://pypi.org/simple "paddlepaddle-gpu==3.3.0.dev20251204"
51
source/install/uv_with_retry.sh pip install --system -v -e .[gpu,test,lmp,cu12,torch,jax] mpi4py --reinstall-package deepmd-kit
52
# See https://github.com/jax-ml/jax/issues/29042
53
- source/install/uv_with_retry.sh pip install -U nvidia-cublas-cu12>=12.9.0.13
+ source/install/uv_with_retry.sh pip install --system -U 'nvidia-cublas-cu12>=12.9.0.13'
54
env:
55
DP_VARIANT: cuda
56
DP_ENABLE_NATIVE_OPTIMIZATION: 1
0 commit comments