File tree Expand file tree Collapse file tree 1 file changed +4
-1
lines changed Expand file tree Collapse file tree 1 file changed +4
-1
lines changed Original file line number Diff line number Diff line change @@ -266,7 +266,10 @@ else
266
266
. ./switch_cuda_version.sh " $desired_cuda "
267
267
# TODO, simplify after anaconda fixes their cudatoolkit versioning inconsistency.
268
268
# see: https://github.com/conda-forge/conda-forge.github.io/issues/687#issuecomment-460086164
269
- if [[ " $desired_cuda " == " 11.7" ]]; then
269
+ if [[ " $desired_cuda " == " 11.8" ]]; then
270
+ export CONDA_CUDATOOLKIT_CONSTRAINT=" - pytorch-cuda >=11.9,<12.0 # [not osx]"
271
+ export MAGMA_PACKAGE=" - magma-cuda118 # [not osx and not win]"
272
+ elif [[ " $desired_cuda " == " 11.7" ]]; then
270
273
export CONDA_CUDATOOLKIT_CONSTRAINT=" - pytorch-cuda >=11.7,<11.8 # [not osx]"
271
274
export MAGMA_PACKAGE=" - magma-cuda117 # [not osx and not win]"
272
275
elif [[ " $desired_cuda " == " 11.6" ]]; then
You can’t perform that action at this time.
0 commit comments