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

[skip ci] update to cuda 10.1.2

parent 91903d3e
......@@ -17,7 +17,7 @@ jobs:
# torch-version: [1.8.0, 1.9.0]
torch-version: [1.8.0]
# cuda-version: ['cpu', 'cu101', 'cu102', 'cu111']
cuda-version: ['cu101', 'cu102']
cuda-version: ['cu101']
exclude:
- torch-version: 1.9.0
cuda-version: 'cu101'
......@@ -48,6 +48,8 @@ jobs:
run: |
pip install numpy typing-extensions
pip install --no-index --no-cache-dir torch==${{ matrix.torch-version}} -f https://download.pytorch.org/whl/${{ matrix.cuda-version }}/torch_stable.html
python -c "import torch; print('PyTorch:', torch.__version__)"
python -c "import torch; print('CUDA:', torch.version.cuda)"
- name: Install main package for CPU
if: ${{ matrix.cuda-version == 'cpu' }}
......
......@@ -7,7 +7,7 @@ curl -k -L "https://drive.google.com/u/0/uc?id=1injUyo3lnarMgWyRcXqKg4UGnN0ysmuq
export CUDA_SHORT=10.1
export CUDA_URL=https://developer.nvidia.com/compute/cuda/${CUDA_SHORT}/Prod/local_installers
export CUDA_FILE=cuda_${CUDA_SHORT}.105_418.96_win10.exe
export CUDA_FILE=cuda_${CUDA_SHORT}.243_426.00_win10.exe
# Install CUDA:
curl -k -L "${CUDA_URL}/${CUDA_FILE}" --output "${CUDA_FILE}"
......
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