wrong string
parent
5d9d527ec5
commit
7e8f65ee22
|
@ -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'
|
||||
|
|
Loading…
Reference in New Issue