Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Support torch 1.11.0 and CUDA 11.5 #931

Merged
merged 7 commits into from
Mar 15, 2022
Merged

Support torch 1.11.0 and CUDA 11.5 #931

Show file tree
Hide file tree
Changes from 1 commit
Commits
Show all changes
7 commits
Select commit Hold shift + click to select a range
975700c
Support torch 1.11.0 and CUDA 11.5
csukuangfj Mar 11, 2022
1bfcbdf
Minor fixes.
csukuangfj Mar 11, 2022
18ccdcc
Minor fixes.
csukuangfj Mar 11, 2022
c8c2417
Minor fixes.
csukuangfj Mar 12, 2022
17c4b51
Minor fixes.
csukuangfj Mar 12, 2022
20ba002
Fix a typo.
csukuangfj Mar 14, 2022
eee0761
minor fixes.
csukuangfj Mar 14, 2022
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Next Next commit
Support torch 1.11.0 and CUDA 11.5
  • Loading branch information
@csukuangfj
csukuangfj committed Mar 11, 2022
commit 975700cdeaabb466ce3e5a6c3f5189225bc616f3
31 changes: 28 additions & 3 deletions .github/workflows/build-cpu.yml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -36,10 +36,33 @@ jobs:
fail-fast: false
matrix:
os: [ubuntu-18.04, macos-10.15]
torch: ["1.5.0", "1.5.1", "1.6.0", "1.7.0", "1.7.1", "1.8.0", "1.8.1", "1.9.0", "1.9.1", "1.10.0", "1.10.1", "1.10.2"]
# Python 3.9 is for PyTorch 1.7.1, 1.8.x, 1.9.x, 1.10.x
python-version: [3.6, 3.7, 3.8, 3.9]
torch: ["1.5.0", "1.5.1", "1.6.0", "1.7.0", "1.7.1", "1.8.0", "1.8.1", "1.9.0", "1.9.1", "1.10.0", "1.10.1", "1.10.2", "1.11.0"]
# Python 3.9 is for PyTorch 1.7.1, 1.8.x, 1.9.x, 1.10.x, 1.11.x
# Python 3.10 is for PyTorch 1.11.x
python-version: [3.6, 3.7, 3.8, 3.9, 3.10]
exclude:
- python-version: 3.10 # exclude Python 3.9 for [1.5.0, 1.5.1, 1.6.0, 1.7.0, 1.7.1, 1.8.0, 1.8.1, 1.9.0, 1.9.1, 1.10.0, 1.10.1, 1.10.2]
torch: "1.5.0"
- python-version: 3.10
torch: "1.5.1"
- python-version: 3.10
torch: "1.6.0"
- python-version: 3.10
torch: "1.7.0"
- python-version: 3.10
torch: "1.8.0"
- python-version: 3.10
torch: "1.8.1"
- python-version: 3.10
torch: "1.9.0"
- python-version: 3.10
torch: "1.9.1"
- python-version: 3.10
torch: "1.10.0"
- python-version: 3.10
torch: "1.10.1"
- python-version: 3.10
torch: "1.10.2"
- python-version: 3.9 # exclude Python 3.9 for [1.5.0, 1.5.1, 1.6.0, 1.7.0]
torch: "1.5.0"
- python-version: 3.9
Expand All @@ -48,6 +71,8 @@ jobs:
torch: "1.6.0"
- python-version: 3.9
torch: "1.7.0"
- python-version: 3.6 # exclude Python 3.6 for [1.11.0]
torch: "1.11.0"

steps:
# refer to https://github.com/actions/checkout
Expand Down
78 changes: 70 additions & 8 deletions .github/workflows/build.yml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -39,21 +39,51 @@ jobs:
# from https://download.pytorch.org/whl/torch_stable.html
# Note: There are no torch versions for CUDA 11.2
#
# 1.11.x supports: cuda10.2 (default), 11.3, 11.5
# 1.10.x supports: cuda10.2 (default), 11.1, 11.3
# 1.9.x supports: cuda10.2 (default), 11.1
# PyTorch 1.8.x supports: cuda 10.1, 10.2 (default), 11.1
# PyTorch 1.7.x supports: cuda 10.1, 10.2 (default), 11.0
# PyTorch 1.6.0 supports: cuda 10.1, 10.2 (default)
# PyTorch 1.5.x supports: cuda 10.1, 10.2 (default)
# Other PyTorch versions are not tested
# CUDA 11.3 is for torch 1.10
cuda: ["10.1", "10.2", "11.0", "11.1", "11.3"]
# CUDA 10.1 is for 1.5.x, 1.6.0, 1.7.x, 1.8.x
# CUDA 11.1 is for torch 1.8.x, 1.9.x, 1.10.x
# CUDA 11.3 is for torch 1.10, 1.11.x
# CUDA 11.5 is for torch 1.11.x
cuda: ["10.1", "10.2", "11.0", "11.1", "11.3", "11.5"]
gcc: ["7"]
torch: ["1.5.0", "1.5.1", "1.6.0", "1.7.0", "1.7.1", "1.8.0", "1.8.1", "1.9.0", "1.9.1", "1.10.0", "1.10.1", "1.10.2"]
torch: ["1.5.0", "1.5.1", "1.6.0", "1.7.0", "1.7.1", "1.8.0", "1.8.1", "1.9.0", "1.9.1", "1.10.0", "1.10.1", "1.10.2", "1.11.0"]
#
# Python 3.9 is for PyTorch 1.7.1, 1.8.0, 1.8.1, 1.9.x, 1.10.x
python-version: [3.6, 3.7, 3.8, 3.9]
# torch 1.11.x does not support Python 3.6
# From torch 1.11.x, it supports Python 3.10
# Python 3.9 is for PyTorch 1.7.1, 1.8.0, 1.8.1, 1.9.x, 1.10.x, 11.x
python-version: [3.6, 3.7, 3.8, 3.9, 3.10]
exclude:
- cuda: "11.5" # exclude 11.3 for [1.5.0, 1.5.1, 1.6.0, 1.7.0, 1.7.1, 1.8.0, 1.8.1, 1.9.0, 1.9.1, 1.10.0, 1.10.1, 1.10.2]
torch: "1.5.0"
- cuda: "11.5"
torch: "1.5.1"
- cuda: "11.5"
torch: "1.6.0"
- cuda: "11.5"
torch: "1.7.0"
- cuda: "11.5"
torch: "1.7.1"
- cuda: "11.5"
torch: "1.8.0"
- cuda: "11.5"
torch: "1.8.1"
- cuda: "11.5"
torch: "1.9.0"
- cuda: "11.5"
torch: "1.9.1"
- cuda: "11.5"
torch: "1.10.0"
- cuda: "11.5"
torch: "1.10.1"
- cuda: "11.5"
torch: "1.10.2"
- cuda: "11.3" # exclude 11.3 for [1.5.0, 1.5.1, 1.6.0, 1.7.0, 1.7.1, 1.8.0, 1.8.1, 1.9.0, 1.9.1]
torch: "1.5.0"
- cuda: "11.3"
Expand All @@ -72,7 +102,7 @@ jobs:
torch: "1.9.0"
- cuda: "11.3"
torch: "1.9.1"
- cuda: "11.0" # exclude 11.0 for [1.5.0, 1.5.1, 1.6.0, 1.8.0, 1.8.1, 1.9.0, 1.9.1, 1.10.0, 1.10.1, 1.10.2]
- cuda: "11.0" # exclude 11.0 for [1.5.0, 1.5.1, 1.6.0, 1.8.0, 1.8.1, 1.9.0, 1.9.1, 1.10.0, 1.10.1, 1.10.2, 1.11.0]
torch: "1.5.0"
- cuda: "11.0"
torch: "1.5.1"
Expand All @@ -92,7 +122,9 @@ jobs:
torch: "1.10.1"
- cuda: "11.0"
torch: "1.10.2"
- cuda: "11.1" # exclude 11.1 for [1.5.0, 1.5.1, 1.6.0, 1.7.0, 1.7.1]
- cuda: "11.0"
torch: "1.11.0"
- cuda: "11.1" # exclude 11.1 for [1.5.0, 1.5.1, 1.6.0, 1.7.0, 1.7.1, 1.11.0]
torch: "1.5.0"
- cuda: "11.1"
torch: "1.5.1"
Expand All @@ -102,7 +134,9 @@ jobs:
torch: "1.7.0"
- cuda: "11.1"
torch: "1.7.1"
- cuda: "10.1" # exclude CUDA 10.1 for [1.9.0, 1.9.1, 1.10.0, 10.1, 10.2]
- cuda: "11.1"
torch: "1.11.0"
- cuda: "10.1" # exclude CUDA 10.1 for [1.9.0, 1.9.1, 1.10.0, 10.1, 10.2, 1.11.0]
torch: "1.9.0"
- cuda: "10.1"
torch: "1.9.1"
Expand All @@ -112,6 +146,8 @@ jobs:
torch: "1.10.1"
- cuda: "10.1"
torch: "1.10.2"
- cuda: "10.1"
torch: "1.11.0"
- python-version: 3.9 # exclude Python 3.9 for [1.5.0, 1.5.1, 1.6.0, 1.7.0]
torch: "1.5.0"
- python-version: 3.9
Expand All @@ -120,6 +156,32 @@ jobs:
torch: "1.6.0"
- python-version: 3.9
torch: "1.7.0"
- python-version: 3.10 # exclude Python 3.9 for [1.5.0, 1.5.1, 1.6.0, 1.7.0, 1.7.1, 1.8.0, 1.8.1, 1.9.0, 1.9.1, 1.10.0, 1.10.1, 1.10.2]
torch: "1.5.0"
- python-version: 3.10
torch: "1.5.1"
- python-version: 3.10
torch: "1.6.0"
- python-version: 3.10
torch: "1.7.0"
- python-version: 3.10
torch: "1.7.1"
- python-version: 3.10
torch: "1.8.0"
- python-version: 3.10
torch: "1.8.1"
- python-version: 3.10
torch: "1.9.0"
- python-version: 3.10
torch: "1.9.1"
- python-version: 3.10
torch: "1.10.0"
- python-version: 3.10
torch: "1.10.1"
- python-version: 3.10
torch: "1.10.2"
- python-version: 3.6 # exclude Python 3.6 for [1.11.0]
torch: "1.11.0"

steps:
# refer to https://github.com/actions/checkout
Expand Down
74 changes: 67 additions & 7 deletions .github/workflows/build_conda.yml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -21,6 +21,7 @@ name: build_conda_cuda
on:
push:
branches:
- master
- conda-cuda

env:
Expand All @@ -33,10 +34,12 @@ jobs:
fail-fast: false
matrix:
os: [ubuntu-18.04]
python-version: [3.6, 3.7, 3.8, 3.9]
cuda: ["10.1", "10.2", "11.0", "11.1", "11.3"]
python-version: [3.6, 3.7, 3.8, 3.9, 3.10]
cuda: ["10.1", "10.2", "11.0", "11.1", "11.3", "11.5"]
# from https://download.pytorch.org/whl/torch_stable.html
# Note: There are no torch versions for CUDA 11.2
#
# 1.11.x supports: cuda10.2 (default), 11.3, 11.5
# PyTorch 1.10.x supports: 10.2 (default), 11.1, 11.3
# PyTorch 1.9.x supports: 10.2 (default), 11.1
# PyTorch 1.8.1 supports: cuda 10.1, 10.2 (default), 11.1
Expand All @@ -45,7 +48,8 @@ jobs:
# PyTorch 1.6.0 supports: cuda 10.1, 10.2 (default), 9.2 (not included in this setup)
# PyTorch 1.5.x supports: cuda 10.1, 10.2 (default), 9.2 (not included in this setup)
#
# PyTorch 1.7.1, 1.8.x, 1.9.x, and 1.10 support 3.6, 3.7, 3.8, 3.9
# PyTorch 1.11.x supports Python 3.10
# PyTorch 1.7.1, 1.8.x, 1.9.x, 1.10.x, and 1.11.x support 3.6, 3.7, 3.8, 3.9
# PyTorch 1.7.0, 1.6.0, and 1.5.x support 3.6, 3.7, 3.8
#
# Other PyTorch versions are not tested
Expand All @@ -56,9 +60,33 @@ jobs:
# https://github.com/csukuangfj/k2/runs/2533830771?check_suite_focus=true
# and
# https://github.com/NVIDIA/apex/issues/805
torch: ["1.6.0", "1.7.0", "1.7.1", "1.8.0", "1.8.1", "1.9.0", "1.9.1", "1.10.0", "1.10.1", "1.10.2"]
torch: ["1.6.0", "1.7.0", "1.7.1", "1.8.0", "1.8.1", "1.9.0", "1.9.1", "1.10.0", "1.10.1", "1.10.2", "1.11.0"]
exclude:
# - cuda: "11.0" # exclude 11.0 for [1.5.0, 1.5.1, 1.6.0, 1.8.0, 1.8.1, 1.9.0, 1.9.1, 1.10.0, 1.10.1, 1.10.2]
- cuda: "11.5" # exclude cuda 11.5 for [1.5.0, 1.5.1, 1.6.0, 1.7.0, 1.7.1, 1.8.0, 1.8.1, 1.9.0, 1.9.1, 1.10.0, 1.10.1, 1.10.2]
torch: "1.5.0"
- cuda: "11.5"
torch: "1.5.1"
- cuda: "11.5"
torch: "1.6.0"
- cuda: "11.5"
torch: "1.7.0"
- cuda: "11.5"
torch: "1.7.1"
- cuda: "11.5"
torch: "1.8.0"
- cuda: "11.5"
torch: "1.8.1"
- cuda: "11.5"
torch: "1.9.0"
- cuda: "11.5"
torch: "1.9.1"
- cuda: "11.5"
torch: "1.10.0"
- cuda: "11.5"
torch: "1.10.1"
- cuda: "11.5"
torch: "1.10.2"
# - cuda: "11.0" # exclude 11.0 for [1.5.0, 1.5.1, 1.6.0, 1.8.0, 1.8.1, 1.9.0, 1.9.1, 1.10.0, 1.10.1, 1.10.2, 1.11.0]
# torch: "1.5.0"
# - cuda: "11.0"
# torch: "1.5.1"
Expand All @@ -78,7 +106,9 @@ jobs:
torch: "1.10.1"
- cuda: "11.0"
torch: "1.10.2"
# - cuda: "11.1" # exclude 11.1 for [1.5.0, 1.5.1, 1.6.0, 1.7.0, 1.7.1]
- cuda: "11.0"
torch: "1.11.0"
# - cuda: "11.1" # exclude 11.1 for [1.5.0, 1.5.1, 1.6.0, 1.7.0, 1.7.1, 1.11.0]
# torch: "1.5.0"
# - cuda: "11.1"
# torch: "1.5.1"
Expand All @@ -88,7 +118,9 @@ jobs:
torch: "1.7.0"
- cuda: "11.1"
torch: "1.7.1"
- cuda: "10.1" # exclude 10.1 for [1.9.0, 1.9.1, 1.10.0, 1.10.1, 1.10.2]
- cuda: "11.1"
torch: "1.11.0"
- cuda: "10.1" # exclude 10.1 for [1.9.0, 1.9.1, 1.10.0, 1.10.1, 1.10.2, 1.11.0]
torch: "1.9.0"
- cuda: "10.1"
torch: "1.9.1"
Expand All @@ -98,6 +130,8 @@ jobs:
torch: "1.10.1"
- cuda: "10.1"
torch: "1.10.2"
- cuda: "10.1"
torch: "1.11.0"
- python-version: 3.9 # exclude Python 3.9 for [1.5.0, 1.5.1, 1.6.0, 1.7.0]
torch: "1.5.0"
- python-version: 3.9
Expand All @@ -106,6 +140,32 @@ jobs:
torch: "1.6.0"
- python-version: 3.9
torch: "1.7.0"
- python-version: 3.10 # exclude Python 3.10 for [1.5.0, 1.5.1, 1.6.0, 1.7.0, 1.7.1, 1.8.0, 1.8.1, 1.9.0, 1.9.1, 1.10.0, 1.10.1, 1.10.2]
torch: "1.5.0"
- python-version: 3.10
torch: "1.5.1"
- python-version: 3.10
torch: "1.6.0"
- python-version: 3.10
torch: "1.7.0"
- python-version: 3.10
torch: "1.7.1"
- python-version: 3.10
torch: "1.8.0"
- python-version: 3.10
torch: "1.8.1"
- python-version: 3.10
torch: "1.9.0"
- python-version: 3.10
torch: "1.9.1"
- python-version: 3.10
torch: "1.10.0"
- python-version: 3.10
torch: "1.10.1"
- python-version: 3.10
torch: "1.10.2"
- python-version: 3.6 # exclude Python 3.6 for [1.11.0]
torch: "1.11.0"

steps:
# refer to https://github.com/actions/checkout
Expand Down
32 changes: 30 additions & 2 deletions .github/workflows/build_conda_cpu.yml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -31,6 +31,7 @@ name: build_conda_cpu
on:
push:
branches:
- master
- conda-cpu

env:
Expand All @@ -43,15 +44,16 @@ jobs:
fail-fast: false
matrix:
os: [ubuntu-18.04, macos-10.15]
python-version: [3.6, 3.7, 3.8, 3.9]
python-version: [3.6, 3.7, 3.8, 3.9, 3.10]
# from https://download.pytorch.org/whl/torch_stable.html
#
# PyTorch 1.11.x supports 3.7, 3.8, 3.9, 3.10
# PyTorch 1.10, 1.9.x, 1.8.x, and 1.7.1 support 3.6, 3.7, 3.8, 3.9
# PyTorch 1.7.0, 1.6.0, and 1.5.x support 3.6, 3.7, 3.8
#
# Other PyTorch versions are not tested
#
torch: ["1.5.0", "1.5.1", "1.6.0", "1.7.0", "1.7.1", "1.8.0", "1.8.1", "1.9.0", "1.9.1", "1.10.0", "1.10.1", "1.10.2"]
torch: ["1.5.0", "1.5.1", "1.6.0", "1.7.0", "1.7.1", "1.8.0", "1.8.1", "1.9.0", "1.9.1", "1.10.0", "1.10.1", "1.10.2", "1.11.0"]
exclude:
- python-version: 3.9 # exclude Python 3.9 for [1.5.0, 1.5.1, 1.6.0, 1.7.0]
torch: "1.5.0"
Expand All @@ -61,6 +63,32 @@ jobs:
torch: "1.6.0"
- python-version: 3.9
torch: "1.7.0"
- python-version: 3.10 # exclude Python 3.10 for [1.5.0, 1.5.1, 1.6.0, 1.7.0, 1.7.1, 1.8.0, 1.8.1, 1.9.0, 1.9.1, 1.10.0, 1.10.1, 1.10.2]
torch: "1.5.0"
- python-version: 3.10
torch: "1.5.1"
- python-version: 3.10
torch: "1.6.0"
- python-version: 3.10
torch: "1.7.0"
- python-version: 3.10
torch: "1.7.1"
- python-version: 3.10
torch: "1.8.0"
- python-version: 3.10
torch: "1.8.1"
- python-version: 3.10
torch: "1.9.0"
- python-version: 3.10
torch: "1.9.1"
- python-version: 3.10
torch: "1.10.0"
- python-version: 3.10
torch: "1.10.1"
- python-version: 3.10
torch: "1.10.2"
- python-version: 3.6 # exclude Python 3.6 for [1.11.0]
torch: "1.11.0"

steps:
# refer to https://github.com/actions/checkout
Expand Down
Loading