Skip to content

Commit

Permalink
[ci skip]
Browse files Browse the repository at this point in the history
  • Loading branch information
rusty1s committed Mar 11, 2022
1 parent 5f95c8e commit faa1287
Show file tree
Hide file tree
Showing 2 changed files with 5 additions and 4 deletions.
5 changes: 3 additions & 2 deletions .github/workflows/building.yml
Original file line number Diff line number Diff line change
Expand Up @@ -15,11 +15,12 @@ jobs:
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: ['cpu', 'cu102', 'cu111', 'cu113', 'cu115']
cuda-version: ['cu115']
# cuda-version: ['cpu', 'cu102', 'cu111', 'cu113', 'cu115']
exclude:
- torch-version: 1.11.0
cuda-version: 'cu111'
- os: windows-latest
- os: windows-2019
torch-version: 1.11.0
cuda-version: 'cu102'
- os: macos-10.15
Expand Down
4 changes: 2 additions & 2 deletions .github/workflows/cuda/cu115-Windows.sh
Original file line number Diff line number Diff line change
Expand Up @@ -2,8 +2,8 @@

# 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"
# 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
Expand Down

0 comments on commit faa1287

Please sign in to comment.