first

Dockerfile

@@ -1,9 +1,8 @@
-FROM nvidia/cuda:11.2.1-runtime-ubuntu20.04
+FROM nvidia/cuda:12.1.1-cudnn8-devel-ubuntu22.04
 
 ARG DEBIAN_FRONTEND=noninteractive
 
 ENV PYTHONUNBUFFERED=1
-ENV CUDA_VISIBLE_DEVICES=1
 
 RUN apt-get update && apt-get install --no-install-recommends -y \
   build-essential \
@@ -15,11 +14,9 @@ RUN apt-get update && apt-get install --no-install-recommends -y \
 
 WORKDIR /code
 
-COPY ./pre-requirements.txt /code/pre-requirements.txt
-
+COPY ./requirements.txt /code/pre-requirements.txt
 COPY ./requirements.txt /code/requirements.txt
 
-
 # Set up a new user named "user" with user ID 1000
 RUN useradd -m -u 1000 user
 # Switch to the "user" user