Commit 2d792ffd authored by rusty1s's avatar rusty1s
Browse files

update

parent 11011409
...@@ -12,28 +12,24 @@ jobs: ...@@ -12,28 +12,24 @@ 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'] python-version: ['3.7', '3.8', '3.9', '3.10', '3.11']
torch-version: [1.13.0] # [1.12.0, 1.13.0] torch-version: [1.13.0, 2.0.0]
cuda-version: ['cpu', 'cu102', 'cu113', 'cu116', 'cu117'] cuda-version: ['cpu', 'cu116', 'cu117', 'cu118']
exclude: exclude:
- torch-version: 1.12.0
cuda-version: 'cu117'
- torch-version: 1.13.0 - torch-version: 1.13.0
cuda-version: 'cu102' python-version: '3.11'
- torch-version: 2.0.0
python-version: '3.7'
- torch-version: 1.13.0 - torch-version: 1.13.0
cuda-version: 'cu113' cuda-version: 'cu118'
- os: macos-10.15 - torch-version: 2.0.0
cuda-version: 'cu102' cuda-version: 'cu116'
- os: macos-10.15
cuda-version: 'cu113'
- os: macos-10.15 - os: macos-10.15
cuda-version: 'cu116' cuda-version: 'cu116'
- os: macos-10.15 - os: macos-10.15
cuda-version: 'cu117' cuda-version: 'cu117'
- os: windows-2019 - os: macos-10.15
cuda-version: 'cu102' cuda-version: 'cu118'
- os: windows-2019 # Complains about CUDA mismatch.
python-version: '3.7'
steps: steps:
- uses: actions/checkout@v2 - uses: actions/checkout@v2
......
``` ```
./build_conda.sh 3.9 1.13.0 cu116 # python, pytorch and cuda version ./build_conda.sh 3.9 2.0.0 cu117 # python, pytorch and cuda version
``` ```
...@@ -10,6 +10,9 @@ if [ "${CUDA_VERSION}" = "cpu" ]; then ...@@ -10,6 +10,9 @@ if [ "${CUDA_VERSION}" = "cpu" ]; then
export CONDA_CUDATOOLKIT_CONSTRAINT="cpuonly # [not osx]" export CONDA_CUDATOOLKIT_CONSTRAINT="cpuonly # [not osx]"
else else
case $CUDA_VERSION in case $CUDA_VERSION in
cu118)
export CONDA_CUDATOOLKIT_CONSTRAINT="pytorch-cuda==11.8.*"
;;
cu117) cu117)
export CONDA_CUDATOOLKIT_CONSTRAINT="pytorch-cuda==11.7.*" export CONDA_CUDATOOLKIT_CONSTRAINT="pytorch-cuda==11.7.*"
;; ;;
......
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