Update Dockerfile and requirements.txt

Dockerfile

@@ -4,12 +4,6 @@ FROM python:3.11.1
 # Set the working directory in the container
 WORKDIR /app
 
-# Run the container as root
-# USER root
-
-# Change permissions of the working directory
-RUN chmod 777 /app  # Gives all users read, write, and exec permissions in the app directory.
-
 # Copy the current directory contents into the container at /usr/src/app
 COPY . .
 
@@ -32,6 +26,12 @@ HEALTHCHECK CMD curl --fail http://localhost:8501/_stcore/health
 # Set the permissions for the script file
 # RUN chmod +x python 3 train_llm.py
 
+# Run the container as root
+# USER root
+
+# Change permissions of the working directory
+RUN chmod 777 /app  # Gives all users read, write, and exec permissions in the app directory.
+
 # Run the command inside your image filesystem.
 CMD ["python", "train_llm.py"]
 

requirements.txt

@@ -108,18 +108,17 @@ nest-asyncio==1.5.8
 networkx==3.2.1
 nltk==3.8.1
 numpy==1.25.2
-nvidia-cublas-cu12==12.1.3.1
-nvidia-cuda-cupti-cu12==12.1.105
-nvidia-cuda-nvrtc-cu12==12.1.105
-nvidia-cuda-runtime-cu12==12.1.105
-nvidia-cudnn-cu12==8.9.2.26
-nvidia-cufft-cu12==11.0.2.54
-nvidia-curand-cu12==10.3.2.106
-nvidia-cusolver-cu12==11.4.5.107
-nvidia-cusparse-cu12==12.1.0.106
-nvidia-nccl-cu12==2.18.1
-nvidia-nvjitlink-cu12==12.3.101
-nvidia-nvtx-cu12==12.1.105
+# nvidia-cuda-cupti-cu12==12.1.105
+# nvidia-cuda-nvrtc-cu12==12.1.105
+# nvidia-cuda-runtime-cu12==12.1.105
+# nvidia-cudnn-cu12==8.9.2.26
+# nvidia-cufft-cu12==11.0.2.54
+# nvidia-curand-cu12==10.3.2.106
+# nvidia-cusolver-cu12==11.4.5.107
+# nvidia-cusparse-cu12==12.1.0.106
+# nvidia-nccl-cu12==2.18.1
+# nvidia-nvjitlink-cu12==12.3.101
+# nvidia-nvtx-cu12==12.1.105
 oauthlib==3.2.2
 openai==1.3.7
 opencv-python==4.8.1.78
@@ -206,7 +205,7 @@ tornado==6.4
 tqdm==4.65.0
 traitlets==5.14.0
 transformers==4.35.2
-triton==2.1.0
+# triton==2.1.0
 trl==0.7.4
 types-jsonschema==4.20.0.0
 types-python-dateutil==2.8.19.14