File tree 1 file changed +3
-3
lines changed 1 file changed +3
-3
lines changed Original file line number Diff line number Diff line change @@ -4,11 +4,11 @@ conda activate ${ENV_NAME}_pypi
4
4
TEST_SUFFIX=" "
5
5
if [[ ${TORCH_ONLY} == ' true' ]]; then
6
6
TEST_SUFFIX=" --package torchonly"
7
- pip3 install --pre torch --index-url " https://download.pytorch.org/whl/${MATRIX_CHANNEL} /${MATRIX_DESIRED_CUDA} _pypi_cudnn"
7
+ pip3 install --pre torch --extra- index-url " https://download.pytorch.org/whl/${MATRIX_CHANNEL} /${MATRIX_DESIRED_CUDA} _pypi_cudnn"
8
8
else
9
9
if [[ ${MATRIX_CHANNEL} != " release" ]]; then
10
- pip3 install --pre torch --index-url " https://download.pytorch.org/whl/${MATRIX_CHANNEL} /${MATRIX_DESIRED_CUDA} _pypi_cudnn"
11
- pip3 install --pre torchvision torchaudio --index-url " https://download.pytorch.org/whl/${MATRIX_CHANNEL} /${MATRIX_DESIRED_CUDA} "
10
+ pip3 install --pre torch --extra- index-url " https://download.pytorch.org/whl/${MATRIX_CHANNEL} /${MATRIX_DESIRED_CUDA} _pypi_cudnn"
11
+ pip3 install --pre torchvision torchaudio --extra- index-url " https://download.pytorch.org/whl/${MATRIX_CHANNEL} /${MATRIX_DESIRED_CUDA} "
12
12
else
13
13
pip3 install torch torchvision torchaudio
14
14
fi
You can’t perform that action at this time.
0 commit comments