Skip to content

Commit

Permalink
Merge pull request #350 from kmaehashi/remove-cuda11.1
Browse files Browse the repository at this point in the history
Drop support for CUDA 11.1 or earlier
  • Loading branch information
emcastillo authored Jul 5, 2023
2 parents 1cbf616 + c974ac4 commit b6d41a1
Showing 1 changed file with 1 addition and 80 deletions.
81 changes: 1 addition & 80 deletions dist_config.py
Original file line number Diff line number Diff line change
Expand Up @@ -10,7 +10,7 @@
# Key-value of sdist build settings.
# See descriptions of WHEEL_LINUX_CONFIGS for details.
SDIST_CONFIG = {
'image': 'nvidia/cuda:10.2-devel-centos7',
'image': 'nvidia/cuda:11.2.2-devel-centos7',
# This image contains cuDNN and NCCL.
'verify_image': 'nvidia/cuda:11.4.2-cudnn8-devel-{system}',
'verify_systems': ['ubuntu18.04'],
Expand All @@ -34,61 +34,6 @@
# - `system_packages`: a string of depending library names expanded into the
# package manager command.
WHEEL_LINUX_CONFIGS = {
'10.2': {
'name': 'cupy-cuda102',
'kind': 'cuda',
'image': 'cupy/cupy-release-tools:cuda-runfile-10.2.89-centos7',
'libs': [
],
'includes': [
],
'preloads': ['cutensor', 'nccl', 'cudnn'],
'verify_image': 'nvidia/cuda:10.2-runtime-{system}',
'verify_systems': ['ubuntu18.04'],
'system_packages': '',
},
'10.2-jetson': {
'name': 'cupy-cuda102',
'kind': 'cuda',
'arch': 'aarch64',
'platform_version': '10.2',
'image': 'cupy/cupy-release-tools:cuda-10.2-jetson', # NOQA
'libs': [
],
'includes': [
],
'builder_dockerfile': 'Dockerfile.jetson',
'preloads': [], # no extra libraries in Jetson
'verify_image': 'nvcr.io/nvidia/l4t-base:r32.5.0',
'verify_systems': ['default'],
'system_packages': '',
},
'11.0': {
'name': 'cupy-cuda110',
'kind': 'cuda',
'image': 'cupy/cupy-release-tools:cuda-runfile-11.0.2-centos7',
'libs': [
],
'includes': [
],
'preloads': ['cutensor', 'nccl', 'cudnn'],
'verify_image': 'nvidia/cuda:11.0.3-runtime-{system}',
'verify_systems': ['ubuntu18.04'],
'system_packages': '',
},
'11.1': {
'name': 'cupy-cuda111',
'kind': 'cuda',
'image': 'cupy/cupy-release-tools:cuda-runfile-11.1.0-centos7',
'libs': [
],
'includes': [
],
'preloads': ['cutensor', 'nccl', 'cudnn'],
'verify_image': 'nvidia/cuda:11.1.1-runtime-{system}',
'verify_systems': ['ubuntu18.04'],
'system_packages': '',
},
'11.x': {
# CUDA Enhanced Compatibility wheel (for CUDA 11.2~11.x)
'name': 'cupy-cuda11x',
Expand Down Expand Up @@ -209,30 +154,6 @@
# - `cudart_lib`: name of CUDA Runtime DLL
# - `check_version`: a function to check if the CUDA version is correct.
WHEEL_WINDOWS_CONFIGS = {
'10.2': {
'name': 'cupy-cuda102',
'kind': 'cuda',
'libs': [],
'preloads': ['cutensor', 'cudnn'],
'cudart_lib': 'cudart64_102',
'check_version': lambda x: 10020 <= x < 10030,
},
'11.0': {
'name': 'cupy-cuda110',
'kind': 'cuda',
'libs': [],
'preloads': ['cutensor', 'cudnn'],
'cudart_lib': 'cudart64_110',
'check_version': lambda x: 11000 <= x < 11010,
},
'11.1': {
'name': 'cupy-cuda111',
'kind': 'cuda',
'libs': [],
'preloads': ['cutensor', 'cudnn'],
'cudart_lib': 'cudart64_110', # binary compatible between CUDA 11.x
'check_version': lambda x: 11010 <= x < 11020,
},
'11.x': {
# CUDA Enhanced Compatibility wheel (for CUDA 11.2~11.x)
'name': 'cupy-cuda11x',
Expand Down

0 comments on commit b6d41a1

Please sign in to comment.