Skip to content

Commit 5ad0c99

Browse files
authored
add conda builds for CUDA 11.8 (#1205)
1 parent 07d5c13 commit 5ad0c99

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

conda/build_pytorch.sh

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -266,7 +266,10 @@ else
266266
. ./switch_cuda_version.sh "$desired_cuda"
267267
# TODO, simplify after anaconda fixes their cudatoolkit versioning inconsistency.
268268
# 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
270273
export CONDA_CUDATOOLKIT_CONSTRAINT=" - pytorch-cuda >=11.7,<11.8 # [not osx]"
271274
export MAGMA_PACKAGE=" - magma-cuda117 # [not osx and not win]"
272275
elif [[ "$desired_cuda" == "11.6" ]]; then

0 commit comments

Comments
 (0)