Commit 7731823a authored by Woosuk Kwon's avatar Woosuk Kwon
Browse files

Remove cuda 11.8

parent d6cd3cd2
......@@ -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
......
......@@ -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
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