`qts8n/cuda-python`

Dockerfile

@@ -1,7 +1,7 @@
 # FROM nvidia/cuda:12.0.0-cudnn8-devel-ubuntu22.04
-FROM python:3.9
+# FROM python:3.9
 # FROM wallies/python-cuda:3.10-cuda11.6-runtime
-# FROM qts8n/cuda-python
+FROM qts8n/cuda-python
 # RUN sudo apt-get remove --purge '^nvidia-.*'
 # RUN sudo apt-get remove --purge '^libnvidia-.*'
 # RUN sudo apt-get autoremove
@@ -24,7 +24,7 @@ RUN pip install --no-cache-dir --upgrade pip
 COPY --chown=user . $HOME/app
 COPY ./requirements.txt ~/app/requirements.txt
 
-RUN apt-get install nvidia-driver-545.92
+# RUN apt-get install nvidia-driver-545.92
 
 RUN pip install -r requirements.txt
 # RUN lspci | grep NVIDIA