wrong string

numpy2_hack
Amir Sadoughi 2024-08-26 15:29:48 -04:00
parent 5d9d527ec5
commit 7e8f65ee22
1 changed files with 2 additions and 2 deletions

View File

@ -63,9 +63,9 @@ 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 pytorch-cuda=12.4 -c "pytorch>=2" -c nvidia/label/cuda-12.4.0
conda install -y -q "pytorch>=2" pytorch-cuda=12.4 -c pytorch -c nvidia/label/cuda-12.4.0
else
conda install -y -q pytorch -c "pytorch>=2"
conda install -y -q "pytorch>=2" -c pytorch
fi
- name: ROCm - Install dependencies
if: inputs.rocm == 'ON'