ci: delete torch tensorflow transformers

RockChinQ · Sep 26, 2023 · fe4df05 · fe4df05
1 parent a28ae78
commit fe4df05
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/Dockerfile b/Dockerfile
@@ -10,6 +10,8 @@ COPY ./main.py /app/main.py
 COPY ./free_one_api /app/free_one_api
 
 RUN pip install -r requirements.txt
+RUN pip uninstall torch tensorflow transformers -y
+RUN rm -rf /usr/local/lib/python3.10/site-packages/nvidia*
 RUN python main.py
 
 CMD [ "python", "main.py" ]