Commit b129b238 authored by rusty1s's avatar rusty1s
Browse files

update

parent 2d792ffd
...@@ -12,12 +12,11 @@ jobs: ...@@ -12,12 +12,11 @@ jobs:
matrix: matrix:
# We have trouble building for Windows - drop for now. # We have trouble building for Windows - drop for now.
os: [ubuntu-18.04, macos-10.15] # windows-2019 os: [ubuntu-18.04, macos-10.15] # windows-2019
python-version: ['3.7', '3.8', '3.9', '3.10', '3.11'] python-version: ['3.7', '3.8', '3.9', '3.10']
torch-version: [1.13.0, 2.0.0] torch-version: [1.13.0, 2.0.0]
cuda-version: ['cpu', 'cu116', 'cu117', 'cu118'] # cuda-version: ['cpu', 'cu116', 'cu117', 'cu118']
cuda-version: ['cu116']
exclude: exclude:
- torch-version: 1.13.0
python-version: '3.11'
- torch-version: 2.0.0 - torch-version: 2.0.0
python-version: '3.7' python-version: '3.7'
- torch-version: 1.13.0 - torch-version: 1.13.0
......
#!/bin/bash #!/bin/bash
CUDA_HOME=/usr/local/cuda-11.6 export CUDA_HOME=/usr/local/cuda-11.6
LD_LIBRARY_PATH=${CUDA_HOME}/lib64:${LD_LIBRARY_PATH} export LD_LIBRARY_PATH=${CUDA_HOME}/lib64:${LD_LIBRARY_PATH}
PATH=${CUDA_HOME}/bin:${PATH} export PATH=${CUDA_HOME}/bin:${PATH}
export FORCE_CUDA=1 export FORCE_CUDA=1
export TORCH_CUDA_ARCH_LIST="3.5;5.0+PTX;6.0;7.0;7.5;8.0;8.6" export TORCH_CUDA_ARCH_LIST="3.5;5.0+PTX;6.0;7.0;7.5;8.0;8.6"
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