Updated Dockerfile

This commit is contained in:
root 2023-08-07 19:22:22 +00:00
parent 2c933106e5
commit 92b03a6875

View File

@ -1,7 +1,6 @@
# Use the official Python image as the base image
FROM nvidia/cuda:11.0.3-base-ubuntu20.04
# Run update
RUN apt update
@ -23,6 +22,8 @@ RUN pip3 install torch torchvision torchaudio --extra-index-url https://download
RUN pip3 install -r requirements.txt
RUN pip3 install runpod
RUN pip3 install websocket-client
RUN pip3 install webcolors
RUN pip3 install scikit-learn
#Give permission to script
RUN chmod +x ./entrypoint.sh