Remove cuda 11.8

vllm-project · May 7, 2024 · 7731823 · 7731823
1 parent d6cd3cd
commit 7731823
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 3 deletions.
diff --git a/.github/workflows/publish.yml b/.github/workflows/publish.yml
@@ -50,7 +50,7 @@ jobs:
           os: ['ubuntu-20.04']
           python-version: ['3.8', '3.9', '3.10', '3.11']
           pytorch-version: ['2.3.0']  # Must be the most recent version that meets requirements-cuda.txt.
-          cuda-version: ['11.8', '12.1']
+          cuda-version: ['12.1']
 
     steps:
       - name: Checkout

diff --git a/.github/workflows/scripts/build.sh b/.github/workflows/scripts/build.sh
@@ -12,7 +12,5 @@ $python_executable -m pip install wheel packaging
 
 # Limit the number of parallel jobs to avoid OOM
 export MAX_JOBS=2
-# Make sure release wheels are built for the following architectures
-export TORCH_CUDA_ARCH_LIST="8.0 8.6 8.9 9.0+PTX"
 # Build
 $python_executable setup.py bdist_wheel --dist-dir=dist