Add CUDA 11.7 & 11.8 to conda packaging (#76)

pytorch · Mar 30, 2023 · 9c1b9f5 · 9c1b9f5
1 parent 497041b
commit 9c1b9f5
Show file tree

Hide file tree

Showing 4 changed files with 12 additions and 22 deletions.
diff --git a/packaging/conda/variants/cu102.yaml b/packaging/conda/variants/cu102.yaml
diff --git a/packaging/conda/variants/cu113.yaml b/packaging/conda/variants/cu113.yaml
diff --git a/packaging/conda/variants/cu116.yaml → packaging/conda/variants/cu117.yaml b/packaging/conda/variants/cu116.yaml → packaging/conda/variants/cu117.yaml
@@ -1,7 +1,7 @@
 cuda:
-  - 11.6                  # [linux64]
+  - 11.7                  # [linux64]
 cuda_home:
-  - /usr/local/cuda-11.6  # [linux64]
+  - /usr/local/cuda-11.7  # [linux64]
 cudnn:
   - 8.3.2                 # [linux64]
 cxx_compiler_version:

diff --git a/packaging/conda/variants/cu118.yaml b/packaging/conda/variants/cu118.yaml
@@ -0,0 +1,10 @@
+cuda:
+  - 11.8                  # [linux64]
+cuda_home:
+  - /usr/local/cuda-11.8  # [linux64]
+cudnn:
+  - 8.3.2                 # [linux64]
+cxx_compiler_version:
+  - 11.2.0                # [linux64]
+pytorch_variant:
+  - cuda                  # [linux64]