rapids cuda deps from conda-forge

facebookresearch · Nov 22, 2024 · 784a8ea · 784a8ea
1 parent 3c54044
commit 784a8ea
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/.github/actions/build_cmake/action.yml b/.github/actions/build_cmake/action.yml
@@ -62,7 +62,7 @@ runs:
           : # skip torch install via conda, we need to install via pip to get
             #  ROCm-enabled version until it's supported in conda by PyTorch
         elif [ "${{ inputs.gpu }}" = "ON" ]; then
-          conda install -y -q "pytorch<2.5" pytorch-cuda=12.4 -c "nvidia/label/cuda-12.4.0"
+          conda install -y -q "pytorch<2.5" pytorch-cuda=12.4 -c pytorch -c "nvidia/label/cuda-12.4.0"
         else
           conda install -y -q pytorch -c pytorch
         fi