diff --git a/ubuntu/cuda/Dockerfile b/ubuntu/cuda/Dockerfile index b249806..698edf1 100644 --- a/ubuntu/cuda/Dockerfile +++ b/ubuntu/cuda/Dockerfile @@ -9,10 +9,10 @@ USER root ENV DEBIAN_FRONTEND noninteractive RUN \ - wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-ubuntu2004.pin \ - && 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/ /" \ + wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu${ubuntu_version//./}/x86_64/cuda-ubuntu${ubuntu_version//./}.pin \ + && mv cuda-ubuntu${ubuntu_version//./}.pin /etc/apt/preferences.d/cuda-repository-pin-600 \ + && apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu${ubuntu_version//./}/x86_64/3bf863cc.pub \ + && add-apt-repository "deb https://developer.download.nvidia.com/compute/cuda/repos/ubuntu${ubuntu_version//./}/x86_64/ /" \ && apt-get -qq update \ && apt-get -qq install -y --no-install-recommends cuda-${cuda_version} \ && rm -rf /var/lib/apt/lists/* \ diff --git a/ubuntu/cuda/uptodate.yaml b/ubuntu/cuda/uptodate.yaml index 8af46ef..e5ac3e3 100644 --- a/ubuntu/cuda/uptodate.yaml +++ b/ubuntu/cuda/uptodate.yaml @@ -16,6 +16,7 @@ dockerbuild: - "12-1" - "12-2" - "12-3" + - "13-0" ubuntu_version: - "22.04" - "22.04" @@ -30,6 +31,7 @@ dockerbuild: - "22.04" - "22.04" - "22.04" + - "22.04" build_args: cuda_version: