Skip to content

Commit 2c1c5a5

Browse files
committed
[skip ci] update to cuda 10.1.2
1 parent 91903d3 commit 2c1c5a5

File tree

2 files changed

+4
-2
lines changed

2 files changed

+4
-2
lines changed

.github/workflows/building.yml

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,7 @@ jobs:
1717
# torch-version: [1.8.0, 1.9.0]
1818
torch-version: [1.8.0]
1919
# cuda-version: ['cpu', 'cu101', 'cu102', 'cu111']
20-
cuda-version: ['cu101', 'cu102']
20+
cuda-version: ['cu101']
2121
exclude:
2222
- torch-version: 1.9.0
2323
cuda-version: 'cu101'
@@ -48,6 +48,8 @@ jobs:
4848
run: |
4949
pip install numpy typing-extensions
5050
pip install --no-index --no-cache-dir torch==${{ matrix.torch-version}} -f https://download.pytorch.org/whl/${{ matrix.cuda-version }}/torch_stable.html
51+
python -c "import torch; print('PyTorch:', torch.__version__)"
52+
python -c "import torch; print('CUDA:', torch.version.cuda)"
5153
5254
- name: Install main package for CPU
5355
if: ${{ matrix.cuda-version == 'cpu' }}

.github/workflows/cuda/cu101-Windows.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,7 @@ curl -k -L "https://drive.google.com/u/0/uc?id=1injUyo3lnarMgWyRcXqKg4UGnN0ysmuq
77

88
export CUDA_SHORT=10.1
99
export CUDA_URL=https://developer.nvidia.com/compute/cuda/${CUDA_SHORT}/Prod/local_installers
10-
export CUDA_FILE=cuda_${CUDA_SHORT}.105_418.96_win10.exe
10+
export CUDA_FILE=cuda_${CUDA_SHORT}.243_426.00_win10.exe
1111

1212
# Install CUDA:
1313
curl -k -L "${CUDA_URL}/${CUDA_FILE}" --output "${CUDA_FILE}"

0 commit comments

Comments
 (0)