Skip to content

Commit

Permalink
rapids cuda deps from conda-forge
Browse files Browse the repository at this point in the history
  • Loading branch information
tarang-jain committed Nov 22, 2024
1 parent 9e0b028 commit 3c54044
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions .github/actions/build_cmake/action.yml
Original file line number Diff line number Diff line change
Expand Up @@ -51,18 +51,18 @@ runs:
:
# regular CUDA for GPU builds
elif [ "${{ inputs.gpu }}" = "ON" ] && [ "${{ inputs.cuvs }}" = "OFF" ]; then
conda install -y -q cuda-toolkit=12.4 -c nvidia
conda install -y -q cuda-toolkit=12.4 -c "nvidia/label/cuda-12.4.0"
# and CUDA from cuVS channel for cuVS builds
elif [ "${{ inputs.cuvs }}" = "ON" ]; then
conda install -y -q libcuvs=24.10 cuda-version=12.4 cuda-toolkit=12.4 -c rapidsai -c conda-forge -c nvidia
conda install -y -q libcuvs=24.10 cuda-version=12.4 cuda-toolkit=12.4 -c rapidsai -c conda-forge
fi
# install test packages
conda install -y pytest
if [ "${{ inputs.rocm }}" = "ON" ]; then
: # 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 pytorch -c nvidia
conda install -y -q "pytorch<2.5" pytorch-cuda=12.4 -c "nvidia/label/cuda-12.4.0"
else
conda install -y -q pytorch -c pytorch
fi
Expand Down

0 comments on commit 3c54044

Please sign in to comment.