Commit ee75bbfa authored by rusty1s's avatar rusty1s
Browse files

update

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