Commit ee75bbfa authored by rusty1s's avatar rusty1s
Browse files

update

parent 1b57f230
...@@ -31,10 +31,6 @@ jobs: ...@@ -31,10 +31,6 @@ jobs:
cuda-version: 'cu102' cuda-version: 'cu102'
- os: windows-2019 # Complains about CUDA mismatch. - os: windows-2019 # Complains about CUDA mismatch.
python-version: '3.7' python-version: '3.7'
cuda-version: 'cu115'
- os: windows-2019 # Complains about CUDA mismatch.
python-version: '3.7'
cuda-version: 'cu116'
steps: steps:
- uses: actions/checkout@v2 - uses: actions/checkout@v2
......
...@@ -31,10 +31,6 @@ jobs: ...@@ -31,10 +31,6 @@ jobs:
cuda-version: 'cu102' cuda-version: 'cu102'
- os: windows-2019 # Complains about CUDA mismatch. - os: windows-2019 # Complains about CUDA mismatch.
python-version: '3.7' python-version: '3.7'
cuda-version: 'cu115'
- os: windows-2019 # Complains about CUDA mismatch.
python-version: '3.7'
cuda-version: 'cu116'
steps: steps:
- uses: actions/checkout@v2 - uses: actions/checkout@v2
......
...@@ -10,6 +10,9 @@ if [ "${CUDA_VERSION}" = "cpu" ]; then ...@@ -10,6 +10,9 @@ if [ "${CUDA_VERSION}" = "cpu" ]; then
export CONDA_CUDATOOLKIT_CONSTRAINT="cpuonly # [not osx]" export CONDA_CUDATOOLKIT_CONSTRAINT="cpuonly # [not osx]"
else else
case $CUDA_VERSION in case $CUDA_VERSION in
cu116)
export CONDA_CUDATOOLKIT_CONSTRAINT="cudatoolkit==11.6.*"
;;
cu115) cu115)
export CONDA_CUDATOOLKIT_CONSTRAINT="cudatoolkit==11.5.*" export CONDA_CUDATOOLKIT_CONSTRAINT="cudatoolkit==11.5.*"
;; ;;
......
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