File tree Expand file tree Collapse file tree 1 file changed +14
-0
lines changed Expand file tree Collapse file tree 1 file changed +14
-0
lines changed Original file line number Diff line number Diff line change @@ -2,6 +2,7 @@ if [[ ${MATRIX_PACKAGE_TYPE} == "libtorch" ]]; then
2
2
curl ${MATRIX_INSTALLATION} -o libtorch.zip
3
3
unzip libtorch.zip
4
4
else
5
+ # special case for Python 3.11
5
6
if [ $MATRIX_PYTHON_VERSION == ' 3.11' ]; then
6
7
export CPYTHON_VERSIONS=3.11.0
7
8
sudo yum -y install openssl-devel libssl-dev bzip2-devel libffi-devel
15
16
eval ${PIP_INSTALLATION}
16
17
eval ${PYTHON_PATH} /python ./test/smoke_test/smoke_test.py --package torchonly
17
18
else
19
+
20
+ # Special case Pypi installation package, only applicable to linux nightly CUDA 11.7 builds, wheel package
21
+ if [[ ${TARGET_OS} == ' linux' && ${MATRIX_CHANNEL} == ' nightly' && ${MATRIX_GPU_ARCH_VERSION} == ' 11.7' && ${MATRIX_PACKAGE_TYPE} == ' manywheel' ]]; then
22
+ conda create -yp ${ENV_NAME} _pypi python=${MATRIX_PYTHON_VERSION} numpy
23
+ INSTALLATION_PYPI=${MATRIX_INSTALLATION/ " cu117" / " cu117_pypi_cudnn" }
24
+ INSTALLATION_PYPI=${INSTALLATION_PYPI/ " torchvision torchaudio" / " " }
25
+ INSTALLATION_PYPI=${INSTALLATION_PYPI/ " index-url" / " extra-index-url" }
26
+ conda run -p ${ENV_NAME} _pypi ${INSTALLATION_PYPI}
27
+ conda run -p ${ENV_NAME} _pypi python ./test/smoke_test/smoke_test.py --package torchonly
28
+ conda activate base
29
+ conda env remove -p ${ENV_NAME} _pypi
30
+ fi
31
+
18
32
conda create -y -n ${ENV_NAME} python=${MATRIX_PYTHON_VERSION} numpy pillow
19
33
conda activate ${ENV_NAME}
20
34
INSTALLATION=${MATRIX_INSTALLATION/ " conda install" / " conda install -y" }
You can’t perform that action at this time.
0 commit comments