Update Dockerfile with cuda container

infr-ai · Jan 8, 2024 · 02a850d · 02a850d
1 parent 1987928
commit 02a850d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/Dockerfile b/Dockerfile
@@ -1,4 +1,4 @@
-FROM ubuntu:22.04
+FROM nvidia/cuda:12.3.1-devel-ubuntu22.04
 
 ENV DEBIAN_FRONTEND=noninteractive
 RUN apt update