Commit 1b9ec8ee authored by rusty1s's avatar rusty1s
Browse files

[ci skip]

parent 1925438c
...@@ -10,7 +10,8 @@ jobs: ...@@ -10,7 +10,8 @@ jobs:
strategy: strategy:
fail-fast: false fail-fast: false
matrix: matrix:
os: [ubuntu-16.04, macos-10.15, windows-latest] # os: [ubuntu-16.04, macos-10.15, windows-latest]
os: [macos-10.15]
python-version: [3.6, 3.7, 3.8, 3.9] python-version: [3.6, 3.7, 3.8, 3.9]
torch-version: [1.8.0, 1.9.0] torch-version: [1.8.0, 1.9.0]
cuda-version: ['cpu', 'cu101', 'cu102', 'cu111'] cuda-version: ['cpu', 'cu101', 'cu102', 'cu111']
......
...@@ -6,10 +6,8 @@ export CUDA_VERSION=$3 ...@@ -6,10 +6,8 @@ export CUDA_VERSION=$3
export CONDA_PYTORCH_CONSTRAINT="pytorch==${TORCH_VERSION%.*}.*" export CONDA_PYTORCH_CONSTRAINT="pytorch==${TORCH_VERSION%.*}.*"
if [[ "$(uname)" == Darwin ]]; then if [ "${CUDA_VERSION}" = "cpu" ]; then
export CONDA_CUDATOOLKIT_CONSTRAINT="" export CONDA_CUDATOOLKIT_CONSTRAINT="cpuonly # [not osx]"
elif [ "${CUDA_VERSION}" = "cpu" ]; then
export CONDA_CUDATOOLKIT_CONSTRAINT="cpuonly"
else else
case $CUDA_VERSION in case $CUDA_VERSION in
cu111) cu111)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment