Update Dockerfile

WildMeOrg · Jun 14, 2024 · e2c13ff · e2c13ff
1 parent 12e82ae
commit e2c13ff
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/devops/base/Dockerfile b/devops/base/Dockerfile
@@ -1,6 +1,6 @@
-ARG WBIA_UBUNTU_IMAGE=12.1.0-cudnn8-runtime-ubuntu22.04
+ARG WBIA_UBUNTU_IMAGE=nvidia/12.1.0-cudnn8-runtime-ubuntu22.04
 #nvidia/cuda:12.2.2-cudnn8-runtime-ubuntu22.04
-docker pull nvidia/cuda:12.2.2-cudnn8-runtime-ubuntu22.04
+#docker pull nvidia/cuda:12.2.2-cudnn8-runtime-ubuntu22.04
 #nvidia/cuda:12.4.1-cudnn-runtime-ubuntu22.04
 #nvidia/cuda:11.7.1-cudnn8-runtime-ubuntu22.04