File tree 1 file changed +10
-4
lines changed 1 file changed +10
-4
lines changed Original file line number Diff line number Diff line change 1
1
conda create -yp ${ENV_NAME} _pypi python=${MATRIX_PYTHON_VERSION} numpy ffmpeg
2
2
3
- if [[ ${MATRIX_CHANNEL} != " release" ]]; then
3
+ TEST_SUFFIX=" "
4
+ if [[ ${TORCH_ONLY} == ' true' ]]; then
5
+ TEST_SUFFIX=" --package torchonly"
4
6
conda run -p ${ENV_NAME} _pypi pip3 install --pre torch --index-url " https://download.pytorch.org/whl/${MATRIX_CHANNEL} /${MATRIX_DESIRED_CUDA} _pypi_cudnn"
5
- conda run -p ${ENV_NAME} _pypi pip3 install --pre torchvision torchaudio --index-url " https://download.pytorch.org/whl/${MATRIX_CHANNEL} /${MATRIX_DESIRED_CUDA} "
6
7
else
7
- conda run -p ${ENV_NAME} _pypi pip3 install torch torchvision torchaudio
8
+ if [[ ${MATRIX_CHANNEL} != " release" ]]; then
9
+ conda run -p ${ENV_NAME} _pypi pip3 install --pre torch --index-url " https://download.pytorch.org/whl/${MATRIX_CHANNEL} /${MATRIX_DESIRED_CUDA} _pypi_cudnn"
10
+ conda run -p ${ENV_NAME} _pypi pip3 install --pre torchvision torchaudio --index-url " https://download.pytorch.org/whl/${MATRIX_CHANNEL} /${MATRIX_DESIRED_CUDA} "
11
+ else
12
+ conda run -p ${ENV_NAME} _pypi pip3 install torch torchvision torchaudio
13
+ fi
8
14
fi
9
15
10
- conda run -p ${ENV_NAME} _pypi python ./test/smoke_test/smoke_test.py
16
+ conda run -p ${ENV_NAME} _pypi python ./test/smoke_test/smoke_test.py ${TEST_SUFFIX} --runtime-error-check disabled
11
17
conda deactivate
12
18
conda env remove -p ${ENV_NAME} _pypi
You can’t perform that action at this time.
0 commit comments