feat: update cuda image

turintech · Jan 20, 2024 · f4cd0a3 · f4cd0a3
1 parent 465a41e
commit f4cd0a3
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/.github/workflows/gpu-release.yaml b/.github/workflows/gpu-release.yaml
@@ -7,7 +7,7 @@ env:
  K3S_GPU_IMG: ghcr.io/turintech/k3s:v1.27.4-k3s1-cuda
  K3S_TAG: v1.27.4-k3s1
  NVIDIA_CONTAINER_RUNTIME_VERSION: 3.7.0-1
- NVIDIA_IMAGE: nvidia/cuda:11.2.2-base-ubuntu20.04
+ NVIDIA_IMAGE: nvidia/cuda:11.8.0-base-ubuntu22.04
 jobs:
  publish-image:
  runs-on: ubuntu-latest

diff --git a/docs/usage/advanced/cuda/Dockerfile b/docs/usage/advanced/cuda/Dockerfile
@@ -1,13 +1,17 @@
 # https:/k3d-io/k3d/issues/1108
 ARG K3S_TAG="v1.27.4-k3s1"
-ARG NVIDIA_IMAGE="nvidia/cuda:11.2.0-base-ubuntu18.04"
+ARG NVIDIA_IMAGE="nvidia/cuda:11.8.0-base-ubuntu22.04"
 FROM rancher/k3s:$K3S_TAG as k3s
 
 FROM $NVIDIA_IMAGE
 
 ARG NVIDIA_CONTAINER_RUNTIME_VERSION
 ENV NVIDIA_CONTAINER_RUNTIME_VERSION=$NVIDIA_CONTAINER_RUNTIME_VERSION
 
+# # Install NVIDIA Container Runtime nvidia/cuda:11.2.2-base-ubuntu20.04
+# RUN curl -s -L https://nvidia.github.io/nvidia-container-runtime/gpgkey | apt-key add -
+# RUN curl -s -L https://nvidia.github.io/nvidia-container-runtime/ubuntu20.04/nvidia-container-runtime.list | tee /etc/apt/sources.list.d/nvidia-container-runtime.list
+
 RUN apt-get update && \
  apt-get -y install gnupg2 curl nvidia-container-runtime=${NVIDIA_CONTAINER_RUNTIME_VERSION} && \
  chmod 1777 /tmp && \