Skip to content

Commit 96d0a6e

Browse files
committed
remove CUDA 11.7 builds; add 11.8
1 parent 6ccc712 commit 96d0a6e

File tree

3 files changed

+5
-19
lines changed

3 files changed

+5
-19
lines changed

.circleci/config.yml

Lines changed: 3 additions & 3 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

.circleci/regenerate.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -34,9 +34,9 @@ def cmake_workflows(indentation=6):
3434
for device in device_types:
3535
job = {"name": f"cmake_{os_type}_{device}", "python_version": python_version}
3636

37-
job["cu_version"] = "cu117" if device == "gpu" else "cpu"
37+
job["cu_version"] = "cu118" if device == "gpu" else "cpu"
3838
if device == "gpu" and os_type == "linux":
39-
job["wheel_docker_image"] = "pytorch/manylinux-cuda117"
39+
job["wheel_docker_image"] = "pytorch/manylinux-cuda118"
4040
jobs.append({f"cmake_{os_type}_{device}": job})
4141
return indent(indentation, jobs)
4242

packaging/pkg_helpers.bash

Lines changed: 0 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -62,14 +62,6 @@ setup_cuda() {
6262
fi
6363
export TORCH_CUDA_ARCH_LIST="3.5;5.0+PTX;6.0;7.0;7.5;8.0;8.6;9.0"
6464
;;
65-
cu117)
66-
if [[ "$OSTYPE" == "msys" ]]; then
67-
export CUDA_HOME="C:\\Program Files\\NVIDIA GPU Computing Toolkit\\CUDA\\v11.7"
68-
else
69-
export CUDA_HOME=/usr/local/cuda-11.7/
70-
fi
71-
export TORCH_CUDA_ARCH_LIST="3.5;5.0+PTX;6.0;7.0;7.5;8.0;8.6"
72-
;;
7365
cpu)
7466
;;
7567
rocm*)
@@ -266,9 +258,6 @@ setup_conda_cudatoolkit_constraint() {
266258
cu118)
267259
export CONDA_CUDATOOLKIT_CONSTRAINT="- pytorch-cuda=11.8 # [not osx]"
268260
;;
269-
cu117)
270-
export CONDA_CUDATOOLKIT_CONSTRAINT="- pytorch-cuda=11.7 # [not osx]"
271-
;;
272261
cpu)
273262
export CONDA_CUDATOOLKIT_CONSTRAINT=""
274263
export CONDA_BUILD_VARIANT="cpu"
@@ -295,9 +284,6 @@ setup_conda_cudatoolkit_plain_constraint() {
295284
cu118)
296285
export CONDA_CUDATOOLKIT_CONSTRAINT="pytorch-cuda=11.8"
297286
;;
298-
cu117)
299-
export CONDA_CUDATOOLKIT_CONSTRAINT="pytorch-cuda=11.7"
300-
;;
301287
cpu)
302288
export CONDA_CUDATOOLKIT_CONSTRAINT=""
303289
export CONDA_BUILD_VARIANT="cpu"

0 commit comments

Comments
 (0)