Commit b552d407 authored by rusty1s's avatar rusty1s
Browse files

[ci skip]

parent faa12872
......@@ -10,23 +10,19 @@ jobs:
strategy:
fail-fast: false
matrix:
os: [windows-2019]
# 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']
torch-version: [1.11.0]
# torch-version: [1.10.0, 1.11.0]
cuda-version: ['cu115']
# cuda-version: ['cpu', 'cu102', 'cu111', 'cu113', 'cu115']
torch-version: [1.10.0, 1.11.0]
cuda-version: ['cpu', 'cu102', 'cu113', 'cu115']
exclude:
- torch-version: 1.11.0
cuda-version: 'cu111'
- os: windows-2019
torch-version: 1.11.0
cuda-version: 'cu102'
- os: windows-2019
torch-version: 1.11.0
cuda-version: 'cu115'
- os: macos-10.15
cuda-version: 'cu102'
- os: macos-10.15
cuda-version: 'cu111'
- os: macos-10.15
cuda-version: 'cu113'
- os: macos-10.15
......
#!/bin/bash
# Install NVIDIA drivers, see:
# https://github.com/pytorch/vision/blob/master/packaging/windows/internal/cuda_install.bat#L99-L102
# curl -k -L "https://drive.google.com/u/0/uc?id=1injUyo3lnarMgWyRcXqKg4UGnN0ysmuq&export=download" --output "/tmp/gpu_driver_dlls.zip"
# 7z x "/tmp/gpu_driver_dlls.zip" -o"/c/Windows/System32"
export CUDA_SHORT=11.5
export CUDA_URL=https://developer.download.nvidia.com/compute/cuda/${CUDA_SHORT}.2/local_installers
export CUDA_FILE=cuda_${CUDA_SHORT}.2_496.13_windows.exe
......
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