Unverified Commit dc07ac2a authored by Andrey Talman's avatar Andrey Talman Committed by GitHub
Browse files

Removing cuda 102 (#6649)

* Removing cuda 102

Display cuda info

Address comments try to resolve CUDA version issue

More work

Base debugging

Fix cuda version passing

Testing

Adding config.yml

Adding command we use for pytorch vision install

Adding unit tests

Modify install command

Refactor config.in

Move cpu tests to different PR

Remove debug code

Testing similar exception for linux as windows

update test_models.py

Revert "Testing similar exception for linux as windows"

This reverts commit 4aaee0b56376440bb56ecd3efe25b60595a241a5.

Revert "update test_models.py"

This reverts commit 413651a308f68b6801681ddfc2c21201c4df3207.

* Removing debug statement

* Reverting to old command
parent b8af91ab
...@@ -766,7 +766,8 @@ jobs: ...@@ -766,7 +766,8 @@ jobs:
image: ubuntu-2004-cuda-11.4:202110-01 image: ubuntu-2004-cuda-11.4:202110-01
resource_class: gpu.nvidia.medium resource_class: gpu.nvidia.medium
environment: environment:
image_name: "pytorch/manylinux-cuda102" image_name: "pytorch/manylinux-cuda116"
CU_VERSION: << parameters.cu_version >>
PYTHON_VERSION: << parameters.python_version >> PYTHON_VERSION: << parameters.python_version >>
steps: steps:
- checkout - checkout
...@@ -1106,12 +1107,6 @@ workflows: ...@@ -1106,12 +1107,6 @@ workflows:
name: binary_linux_wheel_py3.7_cpu name: binary_linux_wheel_py3.7_cpu
python_version: '3.7' python_version: '3.7'
wheel_docker_image: pytorch/manylinux-cpu wheel_docker_image: pytorch/manylinux-cpu
- binary_linux_wheel:
conda_docker_image: pytorch/conda-builder:cuda102
cu_version: cu102
name: binary_linux_wheel_py3.7_cu102
python_version: '3.7'
wheel_docker_image: pytorch/manylinux-cuda102
- binary_linux_wheel: - binary_linux_wheel:
conda_docker_image: pytorch/conda-builder:cuda116 conda_docker_image: pytorch/conda-builder:cuda116
cu_version: cu116 cu_version: cu116
...@@ -1140,12 +1135,6 @@ workflows: ...@@ -1140,12 +1135,6 @@ workflows:
name: binary_linux_wheel_py3.8_cpu name: binary_linux_wheel_py3.8_cpu
python_version: '3.8' python_version: '3.8'
wheel_docker_image: pytorch/manylinux-cpu wheel_docker_image: pytorch/manylinux-cpu
- binary_linux_wheel:
conda_docker_image: pytorch/conda-builder:cuda102
cu_version: cu102
name: binary_linux_wheel_py3.8_cu102
python_version: '3.8'
wheel_docker_image: pytorch/manylinux-cuda102
- binary_linux_wheel: - binary_linux_wheel:
conda_docker_image: pytorch/conda-builder:cuda116 conda_docker_image: pytorch/conda-builder:cuda116
cu_version: cu116 cu_version: cu116
...@@ -1174,12 +1163,6 @@ workflows: ...@@ -1174,12 +1163,6 @@ workflows:
name: binary_linux_wheel_py3.9_cpu name: binary_linux_wheel_py3.9_cpu
python_version: '3.9' python_version: '3.9'
wheel_docker_image: pytorch/manylinux-cpu wheel_docker_image: pytorch/manylinux-cpu
- binary_linux_wheel:
conda_docker_image: pytorch/conda-builder:cuda102
cu_version: cu102
name: binary_linux_wheel_py3.9_cu102
python_version: '3.9'
wheel_docker_image: pytorch/manylinux-cuda102
- binary_linux_wheel: - binary_linux_wheel:
conda_docker_image: pytorch/conda-builder:cuda116 conda_docker_image: pytorch/conda-builder:cuda116
cu_version: cu116 cu_version: cu116
...@@ -1208,12 +1191,6 @@ workflows: ...@@ -1208,12 +1191,6 @@ workflows:
name: binary_linux_wheel_py3.10_cpu name: binary_linux_wheel_py3.10_cpu
python_version: '3.10' python_version: '3.10'
wheel_docker_image: pytorch/manylinux-cpu wheel_docker_image: pytorch/manylinux-cpu
- binary_linux_wheel:
conda_docker_image: pytorch/conda-builder:cuda102
cu_version: cu102
name: binary_linux_wheel_py3.10_cu102
python_version: '3.10'
wheel_docker_image: pytorch/manylinux-cuda102
- binary_linux_wheel: - binary_linux_wheel:
conda_docker_image: pytorch/conda-builder:cuda116 conda_docker_image: pytorch/conda-builder:cuda116
cu_version: cu116 cu_version: cu116
...@@ -1364,12 +1341,6 @@ workflows: ...@@ -1364,12 +1341,6 @@ workflows:
name: binary_linux_conda_py3.7_cpu name: binary_linux_conda_py3.7_cpu
python_version: '3.7' python_version: '3.7'
wheel_docker_image: pytorch/manylinux-cpu wheel_docker_image: pytorch/manylinux-cpu
- binary_linux_conda:
conda_docker_image: pytorch/conda-builder:cuda102
cu_version: cu102
name: binary_linux_conda_py3.7_cu102
python_version: '3.7'
wheel_docker_image: pytorch/manylinux-cuda102
- binary_linux_conda: - binary_linux_conda:
conda_docker_image: pytorch/conda-builder:cuda116 conda_docker_image: pytorch/conda-builder:cuda116
cu_version: cu116 cu_version: cu116
...@@ -1388,12 +1359,6 @@ workflows: ...@@ -1388,12 +1359,6 @@ workflows:
name: binary_linux_conda_py3.8_cpu name: binary_linux_conda_py3.8_cpu
python_version: '3.8' python_version: '3.8'
wheel_docker_image: pytorch/manylinux-cpu wheel_docker_image: pytorch/manylinux-cpu
- binary_linux_conda:
conda_docker_image: pytorch/conda-builder:cuda102
cu_version: cu102
name: binary_linux_conda_py3.8_cu102
python_version: '3.8'
wheel_docker_image: pytorch/manylinux-cuda102
- binary_linux_conda: - binary_linux_conda:
conda_docker_image: pytorch/conda-builder:cuda116 conda_docker_image: pytorch/conda-builder:cuda116
cu_version: cu116 cu_version: cu116
...@@ -1412,12 +1377,6 @@ workflows: ...@@ -1412,12 +1377,6 @@ workflows:
name: binary_linux_conda_py3.9_cpu name: binary_linux_conda_py3.9_cpu
python_version: '3.9' python_version: '3.9'
wheel_docker_image: pytorch/manylinux-cpu wheel_docker_image: pytorch/manylinux-cpu
- binary_linux_conda:
conda_docker_image: pytorch/conda-builder:cuda102
cu_version: cu102
name: binary_linux_conda_py3.9_cu102
python_version: '3.9'
wheel_docker_image: pytorch/manylinux-cuda102
- binary_linux_conda: - binary_linux_conda:
conda_docker_image: pytorch/conda-builder:cuda116 conda_docker_image: pytorch/conda-builder:cuda116
cu_version: cu116 cu_version: cu116
...@@ -1436,12 +1395,6 @@ workflows: ...@@ -1436,12 +1395,6 @@ workflows:
name: binary_linux_conda_py3.10_cpu name: binary_linux_conda_py3.10_cpu
python_version: '3.10' python_version: '3.10'
wheel_docker_image: pytorch/manylinux-cpu wheel_docker_image: pytorch/manylinux-cpu
- binary_linux_conda:
conda_docker_image: pytorch/conda-builder:cuda102
cu_version: cu102
name: binary_linux_conda_py3.10_cu102
python_version: '3.10'
wheel_docker_image: pytorch/manylinux-cuda102
- binary_linux_conda: - binary_linux_conda:
conda_docker_image: pytorch/conda-builder:cuda116 conda_docker_image: pytorch/conda-builder:cuda116
cu_version: cu116 cu_version: cu116
...@@ -1635,7 +1588,7 @@ workflows: ...@@ -1635,7 +1588,7 @@ workflows:
name: unittest_linux_cpu_py3.10 name: unittest_linux_cpu_py3.10
python_version: '3.10' python_version: '3.10'
- unittest_linux_gpu: - unittest_linux_gpu:
cu_version: cu102 cu_version: cu116
filters: filters:
branches: branches:
only: only:
...@@ -1644,11 +1597,11 @@ workflows: ...@@ -1644,11 +1597,11 @@ workflows:
name: unittest_linux_gpu_py3.7 name: unittest_linux_gpu_py3.7
python_version: '3.7' python_version: '3.7'
- unittest_linux_gpu: - unittest_linux_gpu:
cu_version: cu102 cu_version: cu116
name: unittest_linux_gpu_py3.8 name: unittest_linux_gpu_py3.8
python_version: '3.8' python_version: '3.8'
- unittest_linux_gpu: - unittest_linux_gpu:
cu_version: cu102 cu_version: cu116
filters: filters:
branches: branches:
only: only:
...@@ -1657,7 +1610,7 @@ workflows: ...@@ -1657,7 +1610,7 @@ workflows:
name: unittest_linux_gpu_py3.9 name: unittest_linux_gpu_py3.9
python_version: '3.9' python_version: '3.9'
- unittest_linux_gpu: - unittest_linux_gpu:
cu_version: cu102 cu_version: cu116
filters: filters:
branches: branches:
only: only:
...@@ -1682,7 +1635,7 @@ workflows: ...@@ -1682,7 +1635,7 @@ workflows:
name: unittest_windows_cpu_py3.10 name: unittest_windows_cpu_py3.10
python_version: '3.10' python_version: '3.10'
- unittest_windows_gpu: - unittest_windows_gpu:
cu_version: cu102 cu_version: cu116
filters: filters:
branches: branches:
only: only:
...@@ -1691,11 +1644,11 @@ workflows: ...@@ -1691,11 +1644,11 @@ workflows:
name: unittest_windows_gpu_py3.7 name: unittest_windows_gpu_py3.7
python_version: '3.7' python_version: '3.7'
- unittest_windows_gpu: - unittest_windows_gpu:
cu_version: cu102 cu_version: cu116
name: unittest_windows_gpu_py3.8 name: unittest_windows_gpu_py3.8
python_version: '3.8' python_version: '3.8'
- unittest_windows_gpu: - unittest_windows_gpu:
cu_version: cu102 cu_version: cu116
filters: filters:
branches: branches:
only: only:
...@@ -1704,7 +1657,7 @@ workflows: ...@@ -1704,7 +1657,7 @@ workflows:
name: unittest_windows_gpu_py3.9 name: unittest_windows_gpu_py3.9
python_version: '3.9' python_version: '3.9'
- unittest_windows_gpu: - unittest_windows_gpu:
cu_version: cu102 cu_version: cu116
filters: filters:
branches: branches:
only: only:
...@@ -1813,28 +1766,6 @@ workflows: ...@@ -1813,28 +1766,6 @@ workflows:
requires: requires:
- nightly_binary_linux_wheel_py3.7_cpu - nightly_binary_linux_wheel_py3.7_cpu
subfolder: cpu/ subfolder: cpu/
- binary_linux_wheel:
conda_docker_image: pytorch/conda-builder:cuda102
cu_version: cu102
filters:
branches:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.7_cu102
python_version: '3.7'
wheel_docker_image: pytorch/manylinux-cuda102
- binary_wheel_upload:
context: org-member
filters:
branches:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.7_cu102_upload
requires:
- nightly_binary_linux_wheel_py3.7_cu102
subfolder: cu102/
- binary_linux_wheel: - binary_linux_wheel:
conda_docker_image: pytorch/conda-builder:cuda116 conda_docker_image: pytorch/conda-builder:cuda116
cu_version: cu116 cu_version: cu116
...@@ -1943,28 +1874,6 @@ workflows: ...@@ -1943,28 +1874,6 @@ workflows:
requires: requires:
- nightly_binary_linux_wheel_py3.8_cpu - nightly_binary_linux_wheel_py3.8_cpu
subfolder: cpu/ subfolder: cpu/
- binary_linux_wheel:
conda_docker_image: pytorch/conda-builder:cuda102
cu_version: cu102
filters:
branches:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.8_cu102
python_version: '3.8'
wheel_docker_image: pytorch/manylinux-cuda102
- binary_wheel_upload:
context: org-member
filters:
branches:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.8_cu102_upload
requires:
- nightly_binary_linux_wheel_py3.8_cu102
subfolder: cu102/
- binary_linux_wheel: - binary_linux_wheel:
conda_docker_image: pytorch/conda-builder:cuda116 conda_docker_image: pytorch/conda-builder:cuda116
cu_version: cu116 cu_version: cu116
...@@ -2073,28 +1982,6 @@ workflows: ...@@ -2073,28 +1982,6 @@ workflows:
requires: requires:
- nightly_binary_linux_wheel_py3.9_cpu - nightly_binary_linux_wheel_py3.9_cpu
subfolder: cpu/ subfolder: cpu/
- binary_linux_wheel:
conda_docker_image: pytorch/conda-builder:cuda102
cu_version: cu102
filters:
branches:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.9_cu102
python_version: '3.9'
wheel_docker_image: pytorch/manylinux-cuda102
- binary_wheel_upload:
context: org-member
filters:
branches:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.9_cu102_upload
requires:
- nightly_binary_linux_wheel_py3.9_cu102
subfolder: cu102/
- binary_linux_wheel: - binary_linux_wheel:
conda_docker_image: pytorch/conda-builder:cuda116 conda_docker_image: pytorch/conda-builder:cuda116
cu_version: cu116 cu_version: cu116
...@@ -2203,28 +2090,6 @@ workflows: ...@@ -2203,28 +2090,6 @@ workflows:
requires: requires:
- nightly_binary_linux_wheel_py3.10_cpu - nightly_binary_linux_wheel_py3.10_cpu
subfolder: cpu/ subfolder: cpu/
- binary_linux_wheel:
conda_docker_image: pytorch/conda-builder:cuda102
cu_version: cu102
filters:
branches:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.10_cu102
python_version: '3.10'
wheel_docker_image: pytorch/manylinux-cuda102
- binary_wheel_upload:
context: org-member
filters:
branches:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_wheel_py3.10_cu102_upload
requires:
- nightly_binary_linux_wheel_py3.10_cu102
subfolder: cu102/
- binary_linux_wheel: - binary_linux_wheel:
conda_docker_image: pytorch/conda-builder:cuda116 conda_docker_image: pytorch/conda-builder:cuda116
cu_version: cu116 cu_version: cu116
...@@ -2660,27 +2525,6 @@ workflows: ...@@ -2660,27 +2525,6 @@ workflows:
name: nightly_binary_linux_conda_py3.7_cpu_upload name: nightly_binary_linux_conda_py3.7_cpu_upload
requires: requires:
- nightly_binary_linux_conda_py3.7_cpu - nightly_binary_linux_conda_py3.7_cpu
- binary_linux_conda:
conda_docker_image: pytorch/conda-builder:cuda102
cu_version: cu102
filters:
branches:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_conda_py3.7_cu102
python_version: '3.7'
wheel_docker_image: pytorch/manylinux-cuda102
- binary_conda_upload:
context: org-member
filters:
branches:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_conda_py3.7_cu102_upload
requires:
- nightly_binary_linux_conda_py3.7_cu102
- binary_linux_conda: - binary_linux_conda:
conda_docker_image: pytorch/conda-builder:cuda116 conda_docker_image: pytorch/conda-builder:cuda116
cu_version: cu116 cu_version: cu116
...@@ -2744,27 +2588,6 @@ workflows: ...@@ -2744,27 +2588,6 @@ workflows:
name: nightly_binary_linux_conda_py3.8_cpu_upload name: nightly_binary_linux_conda_py3.8_cpu_upload
requires: requires:
- nightly_binary_linux_conda_py3.8_cpu - nightly_binary_linux_conda_py3.8_cpu
- binary_linux_conda:
conda_docker_image: pytorch/conda-builder:cuda102
cu_version: cu102
filters:
branches:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_conda_py3.8_cu102
python_version: '3.8'
wheel_docker_image: pytorch/manylinux-cuda102
- binary_conda_upload:
context: org-member
filters:
branches:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_conda_py3.8_cu102_upload
requires:
- nightly_binary_linux_conda_py3.8_cu102
- binary_linux_conda: - binary_linux_conda:
conda_docker_image: pytorch/conda-builder:cuda116 conda_docker_image: pytorch/conda-builder:cuda116
cu_version: cu116 cu_version: cu116
...@@ -2828,27 +2651,6 @@ workflows: ...@@ -2828,27 +2651,6 @@ workflows:
name: nightly_binary_linux_conda_py3.9_cpu_upload name: nightly_binary_linux_conda_py3.9_cpu_upload
requires: requires:
- nightly_binary_linux_conda_py3.9_cpu - nightly_binary_linux_conda_py3.9_cpu
- binary_linux_conda:
conda_docker_image: pytorch/conda-builder:cuda102
cu_version: cu102
filters:
branches:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_conda_py3.9_cu102
python_version: '3.9'
wheel_docker_image: pytorch/manylinux-cuda102
- binary_conda_upload:
context: org-member
filters:
branches:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_conda_py3.9_cu102_upload
requires:
- nightly_binary_linux_conda_py3.9_cu102
- binary_linux_conda: - binary_linux_conda:
conda_docker_image: pytorch/conda-builder:cuda116 conda_docker_image: pytorch/conda-builder:cuda116
cu_version: cu116 cu_version: cu116
...@@ -2912,27 +2714,6 @@ workflows: ...@@ -2912,27 +2714,6 @@ workflows:
name: nightly_binary_linux_conda_py3.10_cpu_upload name: nightly_binary_linux_conda_py3.10_cpu_upload
requires: requires:
- nightly_binary_linux_conda_py3.10_cpu - nightly_binary_linux_conda_py3.10_cpu
- binary_linux_conda:
conda_docker_image: pytorch/conda-builder:cuda102
cu_version: cu102
filters:
branches:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_conda_py3.10_cu102
python_version: '3.10'
wheel_docker_image: pytorch/manylinux-cuda102
- binary_conda_upload:
context: org-member
filters:
branches:
only: nightly
tags:
only: /v[0-9]+(\.[0-9]+)*-rc[0-9]+/
name: nightly_binary_linux_conda_py3.10_cu102_upload
requires:
- nightly_binary_linux_conda_py3.10_cu102
- binary_linux_conda: - binary_linux_conda:
conda_docker_image: pytorch/conda-builder:cuda116 conda_docker_image: pytorch/conda-builder:cuda116
cu_version: cu116 cu_version: cu116
......
...@@ -766,7 +766,8 @@ jobs: ...@@ -766,7 +766,8 @@ jobs:
image: ubuntu-2004-cuda-11.4:202110-01 image: ubuntu-2004-cuda-11.4:202110-01
resource_class: gpu.nvidia.medium resource_class: gpu.nvidia.medium
environment: environment:
image_name: "pytorch/manylinux-cuda102" image_name: "pytorch/manylinux-cuda116"
CU_VERSION: << parameters.cu_version >>
PYTHON_VERSION: << parameters.python_version >> PYTHON_VERSION: << parameters.python_version >>
steps: steps:
- checkout - checkout
......
...@@ -32,7 +32,7 @@ def build_workflows(prefix="", filter_branch=None, upload=False, indentation=6, ...@@ -32,7 +32,7 @@ def build_workflows(prefix="", filter_branch=None, upload=False, indentation=6,
for os_type in ["linux", "macos", "win"]: for os_type in ["linux", "macos", "win"]:
python_versions = PYTHON_VERSIONS python_versions = PYTHON_VERSIONS
cu_versions_dict = { cu_versions_dict = {
"linux": ["cpu", "cu102", "cu116", "cu117", "rocm5.1.1", "rocm5.2"], "linux": ["cpu", "cu116", "cu117", "rocm5.1.1", "rocm5.2"],
"win": ["cpu", "cu116", "cu117"], "win": ["cpu", "cu116", "cu117"],
"macos": ["cpu"], "macos": ["cpu"],
} }
...@@ -122,7 +122,6 @@ def upload_doc_job(filter_branch): ...@@ -122,7 +122,6 @@ def upload_doc_job(filter_branch):
manylinux_images = { manylinux_images = {
"cu102": "pytorch/manylinux-cuda102",
"cu116": "pytorch/manylinux-cuda116", "cu116": "pytorch/manylinux-cuda116",
"cu117": "pytorch/manylinux-cuda117", "cu117": "pytorch/manylinux-cuda117",
} }
...@@ -247,7 +246,7 @@ def unittest_workflows(indentation=6): ...@@ -247,7 +246,7 @@ def unittest_workflows(indentation=6):
if device_type == "gpu": if device_type == "gpu":
if python_version != "3.8": if python_version != "3.8":
job["filters"] = gen_filter_branch_tree("main", "nightly") job["filters"] = gen_filter_branch_tree("main", "nightly")
job["cu_version"] = "cu102" job["cu_version"] = "cu116"
else: else:
job["cu_version"] = "cpu" job["cu_version"] = "cpu"
......
...@@ -19,13 +19,9 @@ else ...@@ -19,13 +19,9 @@ else
elif [[ ${#CU_VERSION} -eq 5 ]]; then elif [[ ${#CU_VERSION} -eq 5 ]]; then
CUDA_VERSION="${CU_VERSION:2:2}.${CU_VERSION:4:1}" CUDA_VERSION="${CU_VERSION:2:2}.${CU_VERSION:4:1}"
fi fi
echo "Using CUDA $CUDA_VERSION as determined by CU_VERSION" echo "Using CUDA $CUDA_VERSION as determined by CU_VERSION: ${CU_VERSION} "
version="$(python -c "print('.'.join(\"${CUDA_VERSION}\".split('.')[:2]))")" version="$(python -c "print('.'.join(\"${CUDA_VERSION}\".split('.')[:2]))")"
cudatoolkit="pytorch-cuda=${version}"
cudatoolkit="nvidia::cudatoolkit=${version}"
if [[ "$version" == "11.6" || "$version" == "11.7" ]]; then
cudatoolkit=" pytorch-cuda=${version}"
fi
fi fi
case "$(uname -s)" in case "$(uname -s)" in
......
...@@ -62,14 +62,6 @@ setup_cuda() { ...@@ -62,14 +62,6 @@ setup_cuda() {
fi fi
export TORCH_CUDA_ARCH_LIST="3.5;5.0+PTX;6.0;7.0;7.5;8.0;8.6" export TORCH_CUDA_ARCH_LIST="3.5;5.0+PTX;6.0;7.0;7.5;8.0;8.6"
;; ;;
cu102)
if [[ "$OSTYPE" == "msys" ]]; then
export CUDA_HOME="C:\\Program Files\\NVIDIA GPU Computing Toolkit\\CUDA\\v10.2"
else
export CUDA_HOME=/usr/local/cuda-10.2/
fi
export TORCH_CUDA_ARCH_LIST="3.5;5.0+PTX;6.0;7.0;7.5"
;;
cpu) cpu)
;; ;;
rocm*) rocm*)
...@@ -267,9 +259,6 @@ setup_conda_cudatoolkit_constraint() { ...@@ -267,9 +259,6 @@ setup_conda_cudatoolkit_constraint() {
cu116) cu116)
export CONDA_CUDATOOLKIT_CONSTRAINT="- pytorch-cuda=11.6 # [not osx]" export CONDA_CUDATOOLKIT_CONSTRAINT="- pytorch-cuda=11.6 # [not osx]"
;; ;;
cu102)
export CONDA_CUDATOOLKIT_CONSTRAINT="- cudatoolkit >=10.2,<10.3 # [not osx]"
;;
cpu) cpu)
export CONDA_CUDATOOLKIT_CONSTRAINT="" export CONDA_CUDATOOLKIT_CONSTRAINT=""
export CONDA_BUILD_VARIANT="cpu" export CONDA_BUILD_VARIANT="cpu"
...@@ -296,9 +285,6 @@ setup_conda_cudatoolkit_plain_constraint() { ...@@ -296,9 +285,6 @@ setup_conda_cudatoolkit_plain_constraint() {
cu116) cu116)
export CONDA_CUDATOOLKIT_CONSTRAINT="pytorch-cuda=11.6" export CONDA_CUDATOOLKIT_CONSTRAINT="pytorch-cuda=11.6"
;; ;;
cu102)
export CONDA_CUDATOOLKIT_CONSTRAINT="cudatoolkit=10.2"
;;
cpu) cpu)
export CONDA_CUDATOOLKIT_CONSTRAINT="" export CONDA_CUDATOOLKIT_CONSTRAINT=""
export CONDA_BUILD_VARIANT="cpu" export CONDA_BUILD_VARIANT="cpu"
......
...@@ -23,67 +23,12 @@ set CUDA_VERSION_STR=%CUDA_VER_MAJOR%.%CUDA_VER_MINOR% ...@@ -23,67 +23,12 @@ set CUDA_VERSION_STR=%CUDA_VER_MAJOR%.%CUDA_VER_MINOR%
set CUDNN_FOLDER="cuda" set CUDNN_FOLDER="cuda"
set CUDNN_LIB_FOLDER="lib\x64" set CUDNN_LIB_FOLDER="lib\x64"
if %CUDA_VER% EQU 102 goto cuda102
if %CUDA_VER% EQU 113 goto cuda113
if %CUDA_VER% EQU 116 goto cuda116 if %CUDA_VER% EQU 116 goto cuda116
if %CUDA_VER% EQU 117 goto cuda117 if %CUDA_VER% EQU 117 goto cuda117
echo CUDA %CUDA_VERSION_STR% is not supported echo CUDA %CUDA_VERSION_STR% is not supported
exit /b 1 exit /b 1
:cuda102
if not exist "%SRC_DIR%\temp_build\cuda_10.2.89_441.22_win10.exe" (
curl -k -L https://ossci-windows.s3.amazonaws.com/cuda_10.2.89_441.22_win10.exe --output "%SRC_DIR%\temp_build\cuda_10.2.89_441.22_win10.exe"
if errorlevel 1 exit /b 1
set "CUDA_SETUP_FILE=%SRC_DIR%\temp_build\cuda_10.2.89_441.22_win10.exe"
set "ARGS=nvcc_10.2 cuobjdump_10.2 nvprune_10.2 cupti_10.2 cublas_10.2 cublas_dev_10.2 cudart_10.2 cufft_10.2 cufft_dev_10.2 curand_10.2 curand_dev_10.2 cusolver_10.2 cusolver_dev_10.2 cusparse_10.2 cusparse_dev_10.2 nvgraph_10.2 nvgraph_dev_10.2 npp_10.2 npp_dev_10.2 nvjpeg_10.2 nvjpeg_dev_10.2 nvrtc_10.2 nvrtc_dev_10.2 nvml_dev_10.2"
)
if not exist "%SRC_DIR%\temp_build\cudnn-10.2-windows10-x64-v7.6.5.32.zip" (
curl -k -L https://ossci-windows.s3.amazonaws.com/cudnn-10.2-windows10-x64-v7.6.5.32.zip --output "%SRC_DIR%\temp_build\cudnn-10.2-windows10-x64-v7.6.5.32.zip"
if errorlevel 1 exit /b 1
set "CUDNN_SETUP_FILE=%SRC_DIR%\temp_build\cudnn-10.2-windows10-x64-v7.6.5.32.zip"
)
rem The below only for cu102, if it's used in other version, e.g. cu111, torch.cuda.is_availabe() would be False.
if not exist "%SRC_DIR%\temp_build\gpu_driver_dlls.7z" (
curl -k -L "https://drive.google.com/u/0/uc?id=1injUyo3lnarMgWyRcXqKg4UGnN0ysmuq&export=download" --output "%SRC_DIR%\temp_build\gpu_driver_dlls.zip"
if errorlevel 1 exit /b 1
)
echo Installing GPU driver DLLs
7z x %SRC_DIR%\temp_build\gpu_driver_dlls.zip -aoa -o"C:\Windows\System32"
goto cuda_common
:cuda113
set CUDA_INSTALL_EXE=cuda_11.3.0_465.89_win10.exe
if not exist "%SRC_DIR%\temp_build\%CUDA_INSTALL_EXE%" (
curl -k -L "https://ossci-windows.s3.amazonaws.com/%CUDA_INSTALL_EXE%" --output "%SRC_DIR%\temp_build\%CUDA_INSTALL_EXE%"
if errorlevel 1 exit /b 1
set "CUDA_SETUP_FILE=%SRC_DIR%\temp_build\%CUDA_INSTALL_EXE%"
set "ARGS=thrust_11.3 nvcc_11.3 cuobjdump_11.3 nvprune_11.3 nvprof_11.3 cupti_11.3 cublas_11.3 cublas_dev_11.3 cudart_11.3 cufft_11.3 cufft_dev_11.3 curand_11.3 curand_dev_11.3 cusolver_11.3 cusolver_dev_11.3 cusparse_11.3 cusparse_dev_11.3 npp_11.3 npp_dev_11.3 nvjpeg_11.3 nvjpeg_dev_11.3 nvrtc_11.3 nvrtc_dev_11.3 nvml_dev_11.3"
)
set CUDNN_INSTALL_ZIP=cudnn-windows-x86_64-8.3.2.44_cuda11.5-archive.zip
set CUDNN_FOLDER=cudnn-windows-x86_64-8.3.2.44_cuda11.5-archive
set CUDNN_LIB_FOLDER="lib"
if not exist "%SRC_DIR%\temp_build\%CUDNN_INSTALL_ZIP%" (
curl -k -L "http://s3.amazonaws.com/ossci-windows/%CUDNN_INSTALL_ZIP%" --output "%SRC_DIR%\temp_build\%CUDNN_INSTALL_ZIP%"
if errorlevel 1 exit /b 1
set "CUDNN_SETUP_FILE=%SRC_DIR%\temp_build\%CUDNN_INSTALL_ZIP%"
rem Make sure windows path contains zlib dll
curl -k -L "http://s3.amazonaws.com/ossci-windows/zlib123dllx64.zip" --output "%SRC_DIR%\temp_build\zlib123dllx64.zip"
7z x "%SRC_DIR%\temp_build\zlib123dllx64.zip" -o"%SRC_DIR%\temp_build\zlib"
xcopy /Y "%SRC_DIR%\temp_build\zlib\dll_x64\*.dll" "C:\Windows\System32"
)
goto cuda_common
:cuda116 :cuda116
set CUDA_INSTALL_EXE=cuda_11.6.0_511.23_windows.exe set CUDA_INSTALL_EXE=cuda_11.6.0_511.23_windows.exe
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment