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

Deprecate CUDA 11.6 from CI, move to 11.7 (#7163)

parent 82c51c48
...@@ -855,7 +855,7 @@ jobs: ...@@ -855,7 +855,7 @@ jobs:
executor: executor:
name: windows-gpu name: windows-gpu
environment: environment:
CUDA_VERSION: "11.6" CUDA_VERSION: "11.7"
PYTHON_VERSION: << parameters.python_version >> PYTHON_VERSION: << parameters.python_version >>
steps: steps:
- checkout - checkout
......
...@@ -855,7 +855,7 @@ jobs: ...@@ -855,7 +855,7 @@ jobs:
executor: executor:
name: windows-gpu name: windows-gpu
environment: environment:
CUDA_VERSION: "11.6" CUDA_VERSION: "11.7"
PYTHON_VERSION: << parameters.python_version >> PYTHON_VERSION: << parameters.python_version >>
steps: steps:
- checkout - checkout
......
#!/usr/bin/env bash #!/usr/bin/env bash
unset PYTORCH_VERSION unset PYTORCH_VERSION
...@@ -24,7 +25,7 @@ else ...@@ -24,7 +25,7 @@ else
fi fi
cuda_toolkit_pckg="cudatoolkit" cuda_toolkit_pckg="cudatoolkit"
if [[ $CUDA_VERSION == 11.6 || $CUDA_VERSION == 11.7 ]]; then if [[ $CUDA_VERSION == 11.6 || $CUDA_VERSION == 11.7 || $CUDA_VERSION == 11.8 ]]; then
cuda_toolkit_pckg="pytorch-cuda" cuda_toolkit_pckg="pytorch-cuda"
fi fi
......
...@@ -17,7 +17,7 @@ jobs: ...@@ -17,7 +17,7 @@ jobs:
include: include:
- python-version: "3.8" - python-version: "3.8"
gpu-arch-type: cuda gpu-arch-type: cuda
gpu-arch-version: "11.6" gpu-arch-version: "11.7"
runner: linux.4xlarge.nvidia.gpu runner: linux.4xlarge.nvidia.gpu
fail-fast: false fail-fast: false
uses: pytorch/test-infra/.github/workflows/linux_job.yml@main uses: pytorch/test-infra/.github/workflows/linux_job.yml@main
......
...@@ -17,7 +17,7 @@ jobs: ...@@ -17,7 +17,7 @@ jobs:
strategy: strategy:
matrix: matrix:
python_version: ["3.8"] python_version: ["3.8"]
cuda_arch_version: ["11.6"] cuda_arch_version: ["11.7"]
fail-fast: false fail-fast: false
uses: pytorch/test-infra/.github/workflows/linux_job.yml@main uses: pytorch/test-infra/.github/workflows/linux_job.yml@main
with: with:
......
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