File tree Expand file tree Collapse file tree 2 files changed +6
-4
lines changed Expand file tree Collapse file tree 2 files changed +6
-4
lines changed Original file line number Diff line number Diff line change @@ -9,10 +9,10 @@ USER root
9
9
ENV DEBIAN_FRONTEND noninteractive
10
10
11
11
RUN \
12
- wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/ x86_64/cuda-ubuntu2004 .pin \
13
- && mv cuda-ubuntu2004 .pin /etc/apt/preferences.d/cuda-repository-pin-600 \
14
- && apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004 /x86_64/3bf863cc.pub \
15
- && add-apt-repository "deb https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004 /x86_64/ /" \
12
+ wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu${ubuntu_version//./}/ x86_64/cuda-ubuntu${ubuntu_version//./} .pin \
13
+ && mv cuda-ubuntu${ubuntu_version//./} .pin /etc/apt/preferences.d/cuda-repository-pin-600 \
14
+ && apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu${ubuntu_version//./} /x86_64/3bf863cc.pub \
15
+ && add-apt-repository "deb https://developer.download.nvidia.com/compute/cuda/repos/ubuntu${ubuntu_version//./} /x86_64/ /" \
16
16
&& apt-get -qq update \
17
17
&& apt-get -qq install -y --no-install-recommends cuda-${cuda_version} \
18
18
&& rm -rf /var/lib/apt/lists/* \
Original file line number Diff line number Diff line change @@ -16,6 +16,7 @@ dockerbuild:
16
16
- " 12-1"
17
17
- " 12-2"
18
18
- " 12-3"
19
+ - " 13-0"
19
20
ubuntu_version :
20
21
- " 22.04"
21
22
- " 22.04"
@@ -30,6 +31,7 @@ dockerbuild:
30
31
- " 22.04"
31
32
- " 22.04"
32
33
- " 22.04"
34
+ - " 22.04"
33
35
34
36
build_args :
35
37
cuda_version :
You can’t perform that action at this time.
0 commit comments