add cuda-nvrtc

christian-rauch · Dec 28, 2023 · 93d8002 · 93d8002
1 parent 8e90e73
commit 93d8002
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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