updateRequirements

Dockerfile

@@ -1,7 +1,7 @@
 FROM nvidia/cuda:11.3.1-cudnn8-devel-ubuntu18.04
 FROM python:3.8.8
 
-ENV CUDA_HOME=/usr/local/cuda-11.3.1
+
 ARG DEBIAN_FRONTEND=noninteractive
 
 ENV PYTHONUNBUFFERED=1
@@ -23,8 +23,7 @@ RUN useradd -m -u 1000 user
 # Switch to the "user" user
 USER user
 
-WORKDIR /home/user/local
-RUN ls -a
+ENV CUDA_HOME=/home/user/local/cuda-11.3.1
 # Set home to the user's home directory
 ENV HOME=/home/user \
 	PATH=/home/user/.local/bin:$PATH \