diff --git a/.github/workflows/main.yml b/.github/workflows/main.yml index 6df417c..edadda8 100644 --- a/.github/workflows/main.yml +++ b/.github/workflows/main.yml @@ -35,7 +35,7 @@ jobs: mv cuda-ubuntu2004.pin /etc/apt/preferences.d/cuda-repository-pin-600 apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/3bf863cc.pub add-apt-repository "deb https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/ /" - apt install --no-install-recommends -y libcurand-dev-11-3 libcufft-dev-11-3 libcublas-dev-11-3 cuda-nvcc-11-3 cuda-nvtx-11-3 libcudnn8-dev + apt install --no-install-recommends -y libcurand-dev-11-3 libcufft-dev-11-3 libcublas-dev-11-3 cuda-nvcc-11-3 cuda-nvtx-11-3 cuda-nvrtc-dev-11-3 libcudnn8-dev - name: install CUDA (Ubuntu 22.04) if: ${{ matrix.docker_image == 'ubuntu:22.04' }} @@ -44,7 +44,7 @@ jobs: dpkg -i cuda-keyring_1.1-1_all.deb rm cuda-keyring_1.1-1_all.deb apt update - apt install --no-install-recommends -y libcurand-dev-11-8 libcufft-dev-11-8 libcublas-dev-11-8 cuda-nvcc-11-8 cuda-nvtx-11-8 libcudnn8-dev + apt install --no-install-recommends -y libcurand-dev-11-8 libcufft-dev-11-8 libcublas-dev-11-8 cuda-nvcc-11-8 cuda-nvtx-11-8 cuda-nvrtc-dev-11-8 libcudnn8-dev - uses: actions/checkout@v4