File tree 3 files changed +5
-19
lines changed 3 files changed +5
-19
lines changed Original file line number Diff line number Diff line change @@ -34,9 +34,9 @@ def cmake_workflows(indentation=6):
34
34
for device in device_types :
35
35
job = {"name" : f"cmake_{ os_type } _{ device } " , "python_version" : python_version }
36
36
37
- job ["cu_version" ] = "cu117 " if device == "gpu" else "cpu"
37
+ job ["cu_version" ] = "cu118 " if device == "gpu" else "cpu"
38
38
if device == "gpu" and os_type == "linux" :
39
- job ["wheel_docker_image" ] = "pytorch/manylinux-cuda117 "
39
+ job ["wheel_docker_image" ] = "pytorch/manylinux-cuda118 "
40
40
jobs .append ({f"cmake_{ os_type } _{ device } " : job })
41
41
return indent (indentation , jobs )
42
42
Original file line number Diff line number Diff line change @@ -62,14 +62,6 @@ setup_cuda() {
62
62
fi
63
63
export TORCH_CUDA_ARCH_LIST=" 3.5;5.0+PTX;6.0;7.0;7.5;8.0;8.6;9.0"
64
64
;;
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
- ;;
73
65
cpu)
74
66
;;
75
67
rocm* )
@@ -266,9 +258,6 @@ setup_conda_cudatoolkit_constraint() {
266
258
cu118)
267
259
export CONDA_CUDATOOLKIT_CONSTRAINT=" - pytorch-cuda=11.8 # [not osx]"
268
260
;;
269
- cu117)
270
- export CONDA_CUDATOOLKIT_CONSTRAINT=" - pytorch-cuda=11.7 # [not osx]"
271
- ;;
272
261
cpu)
273
262
export CONDA_CUDATOOLKIT_CONSTRAINT=" "
274
263
export CONDA_BUILD_VARIANT=" cpu"
@@ -295,9 +284,6 @@ setup_conda_cudatoolkit_plain_constraint() {
295
284
cu118)
296
285
export CONDA_CUDATOOLKIT_CONSTRAINT=" pytorch-cuda=11.8"
297
286
;;
298
- cu117)
299
- export CONDA_CUDATOOLKIT_CONSTRAINT=" pytorch-cuda=11.7"
300
- ;;
301
287
cpu)
302
288
export CONDA_CUDATOOLKIT_CONSTRAINT=" "
303
289
export CONDA_BUILD_VARIANT=" cpu"
You can’t perform that action at this time.
0 commit comments