nvida/cuda image

Dockerfile

@@ -1,13 +1,4 @@
-FROM python:3.9
-
-
-
-RUN wget https://nvidia.github.io/nvidia-docker/gpgkey --no-check-certificate
-RUN apt-key add gpgkey
-RUN distribution=$(. /etc/os-release;echo $ID$VERSION_ID)
-RUN curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.list | tee /etc/apt/sources.list.d/nvidia-docker.list
-
-
+FROM nvidia/cuda:12.0.0-cudnn8-devel-ubuntu22.04
 
 # Using the Ubuntu image (our OS)
 # Update package manager (apt-get) 
@@ -15,8 +6,7 @@ RUN curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docke
 # Python and Pip
 RUN apt-get update && apt-get install -y \
     python3-pip \ 
-    wget \
-    nvidia-container-toolkit
+    wget 
 
 WORKDIR /code