diff --git a/Dockerfile b/Dockerfile index 1363956..d38d25e 100644 --- a/Dockerfile +++ b/Dockerfile @@ -1,4 +1,3 @@ -# Alternatively, use a base image with CUDA and cuDNN support FROM nvidia/cuda:12.3.2-cudnn9-runtime-ubuntu22.04 RUN apt-get update && apt-get install -y python3-pip