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