Ubuntu 22.04でのCUDAのインストールエラーを回避。

rigaya · May 11, 2024 · 65f3e9d · 65f3e9d
1 parent 25d9e06
commit 65f3e9d
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/docker/docker_ubuntu1804_cuda10 b/docker/docker_ubuntu1804_cuda10
@@ -39,7 +39,7 @@ RUN apt update \
     && dpkg -i ${CUDA_DEB_NAME} \
     && apt-key add ${CUDA_PUB_PATH} \
     && apt-get update \
-    && apt-get -y install cuda-drivers cuda-compiler-${CUDA_VER_MAJOR}-${CUDA_VER_MINOR} cuda-cudart-dev-${CUDA_VER_MAJOR}-${CUDA_VER_MINOR} cuda-driver-dev-${CUDA_VER_MAJOR}-${CUDA_VER_MINOR} cuda-nvrtc-dev-${CUDA_VER_MAJOR}-${CUDA_VER_MINOR} cuda-curand-dev-${CUDA_VER_MAJOR}-${CUDA_VER_MINOR} cuda-npp-dev-${CUDA_VER_MAJOR}-${CUDA_VER_MINOR} cuda-nvml-dev-${CUDA_VER_MAJOR}-${CUDA_VER_MINOR} \
+    && apt-get -y install cuda \
     && rm -f ${CUDA_DEB_NAME} \
     && apt-get clean \
     && rm -rf /var/lib/apt/lists/* \

diff --git a/docker/docker_ubuntu2004_cuda11 b/docker/docker_ubuntu2004_cuda11
@@ -35,7 +35,7 @@ RUN apt update \
     && dpkg -i ${CUDA_DEB_NAME} \
     && apt-key add ${CUDA_PUB_PATH} \
     && apt-get update \
-    && apt-get -y install cuda-drivers cuda-compiler-${CUDA_VER_MAJOR}-${CUDA_VER_MINOR} cuda-cudart-dev-${CUDA_VER_MAJOR}-${CUDA_VER_MINOR} cuda-driver-dev-${CUDA_VER_MAJOR}-${CUDA_VER_MINOR} cuda-nvrtc-dev-${CUDA_VER_MAJOR}-${CUDA_VER_MINOR} libcurand-dev-${CUDA_VER_MAJOR}-${CUDA_VER_MINOR} libnpp-dev-${CUDA_VER_MAJOR}-${CUDA_VER_MINOR} cuda-nvml-dev-${CUDA_VER_MAJOR}-${CUDA_VER_MINOR} \
+    && apt-get -y install cuda \
     && rm -f ${CUDA_DEB_NAME} \
     && apt-get clean \
     && rm -rf /var/lib/apt/lists/* \

diff --git a/docker/docker_ubuntu2204_cuda11 b/docker/docker_ubuntu2204_cuda11
@@ -35,7 +35,7 @@ RUN apt update \
     && dpkg -i ${CUDA_DEB_NAME} \
     && cp ${CUDA_GPG_PATH}/cuda-*-keyring.gpg /usr/share/keyrings/ \
     && apt-get update \
-    && apt-get -y install cuda-drivers cuda-compiler-${CUDA_VER_MAJOR}-${CUDA_VER_MINOR} cuda-cudart-dev-${CUDA_VER_MAJOR}-${CUDA_VER_MINOR} cuda-driver-dev-${CUDA_VER_MAJOR}-${CUDA_VER_MINOR} cuda-nvrtc-dev-${CUDA_VER_MAJOR}-${CUDA_VER_MINOR} libcurand-dev-${CUDA_VER_MAJOR}-${CUDA_VER_MINOR} libnpp-dev-${CUDA_VER_MAJOR}-${CUDA_VER_MINOR} cuda-nvml-dev-${CUDA_VER_MAJOR}-${CUDA_VER_MINOR} \
+    && apt-get -y install cuda \
     && rm -f ${CUDA_DEB_NAME} \
     && apt-get clean \
     && rm -rf /var/lib/apt/lists/*