Download cudnn for windows.

This commit is contained in:
Fangjun Kuang 2022-06-03 14:41:01 +08:00
parent 69a1347a7d
commit db771f4bca
4 changed files with 24 additions and 12 deletions

View File

@ -19,7 +19,7 @@ name: Nightly tests ubuntu conda cpu
on: on:
push: push:
branches: branches:
- nightly-test - nightly-test-2
schedule: schedule:
# minute (0-59) # minute (0-59)
@ -44,13 +44,13 @@ jobs:
id: set-matrix id: set-matrix
run: | run: |
# outputting for debugging purposes # outputting for debugging purposes
python scripts/github_actions/generate_build_matrix.py python scripts/github_actions/generate_build_matrix.py --test-only-latest-torch
MATRIX=$(python scripts/github_actions/generate_build_matrix.py) MATRIX=$(python scripts/github_actions/generate_build_matrix.py --test-only-latest-torch)
echo "::set-output name=matrix::${MATRIX}" echo "::set-output name=matrix::${MATRIX}"
run_nightly_tests_ubuntu_conda_cpu: run_nightly_tests_ubuntu_conda_cpu:
needs: generate_build_matrix needs: generate_build_matrix
runs-on: ubuntu-18.04 runs-on: ubuntu-latest
strategy: strategy:
fail-fast: false fail-fast: false
matrix: matrix:

View File

@ -19,7 +19,7 @@ name: Nightly tests ubuntu conda cuda
on: on:
push: push:
branches: branches:
- nightly-test - nightly-test-2
schedule: schedule:
# minute (0-59) # minute (0-59)
@ -44,13 +44,13 @@ jobs:
id: set-matrix id: set-matrix
run: | run: |
# outputting for debugging purposes # outputting for debugging purposes
python scripts/github_actions/generate_build_matrix.py --enable-cuda python scripts/github_actions/generate_build_matrix.py --enable-cuda --test-only-latest-torch
MATRIX=$(python scripts/github_actions/generate_build_matrix.py --enable-cuda) MATRIX=$(python scripts/github_actions/generate_build_matrix.py --enable-cuda --test-only-latest-torch)
echo "::set-output name=matrix::${MATRIX}" echo "::set-output name=matrix::${MATRIX}"
run_nightly_tests_ubuntu_conda_cuda: run_nightly_tests_ubuntu_conda_cuda:
needs: generate_build_matrix needs: generate_build_matrix
runs-on: ubuntu-18.04 runs-on: ubuntu-latest
strategy: strategy:
fail-fast: false fail-fast: false
matrix: matrix:

View File

@ -19,7 +19,7 @@ name: Nightly tests windows pip cuda
on: on:
push: push:
branches: branches:
- nightly-test - nightly-test-2
schedule: schedule:
# minute (0-59) # minute (0-59)

View File

@ -41,8 +41,8 @@ jobs:
id: set-matrix id: set-matrix
run: | run: |
# outputting for debugging purposes # outputting for debugging purposes
python scripts/github_actions/generate_build_matrix.py --enable-cuda --for-windows python scripts/github_actions/generate_build_matrix.py --enable-cuda --for-windows --test-only-latest-torch
MATRIX=$(python scripts/github_actions/generate_build_matrix.py --enable-cuda --for-windows) MATRIX=$(python scripts/github_actions/generate_build_matrix.py --enable-cuda --for-windows --test-only-latest-torch)
echo "::set-output name=matrix::${MATRIX}" echo "::set-output name=matrix::${MATRIX}"
run_tests_windows_cuda: run_tests_windows_cuda:
@ -72,17 +72,29 @@ jobs:
# See https://github.com/Jimver/cuda-toolkit/blob/master/src/links/windows-links.ts # See https://github.com/Jimver/cuda-toolkit/blob/master/src/links/windows-links.ts
# for available CUDA versions # for available CUDA versions
- uses: Jimver/cuda-toolkit@v0.2.5 - uses: Jimver/cuda-toolkit@v0.2.7
id: cuda-toolkit id: cuda-toolkit
with: with:
cuda: ${{ matrix.cuda }} cuda: ${{ matrix.cuda }}
- name: Display CUDA version - name: Display CUDA version
shell: bash
run: | run: |
echo "Installed cuda version is: ${{ steps.cuda-toolkit.outputs.cuda }}" echo "Installed cuda version is: ${{ steps.cuda-toolkit.outputs.cuda }}"
echo "Cuda install location: ${{steps.cuda-toolkit.outputs.CUDA_PATH}}" echo "Cuda install location: ${{steps.cuda-toolkit.outputs.CUDA_PATH}}"
nvcc --version nvcc --version
- name: Download cuDNN
shell: bash
run: |
GIT_LFS_SKIP_SMUDGE=1 git clone https://huggingface.co/csukuangfj/cudnn-for-windows
cd cudnn-for-windows
git lfs pull --include="cudnn-windows-x86_64-8.4.1.50_cuda11.6-archive.zip"
unzip cudnn-windows-x86_64-8.4.1.50_cuda11.6-archive.zip
ls -lh *
ls -lh */*
- name: Install PyTorch ${{ matrix.torch }} - name: Install PyTorch ${{ matrix.torch }}
shell: bash shell: bash
run: | run: |