roll-ai commited on
Commit
5a8f5b2
·
verified ·
1 Parent(s): c175f6f

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +2 -0
Dockerfile CHANGED
@@ -29,6 +29,8 @@ COPY requirements.txt /app/requirements.txt
29
  RUN conda create -n realcam python=3.10 -y && \
30
  conda run -n realcam pip install --upgrade pip && \
31
  conda run -n realcam conda install ffmpeg=7 -c conda-forge && \
 
 
32
  conda run -n realcam pip install -r /app/requirements.txt
33
 
34
  RUN chmod -R 777 /app /workspace
 
29
  RUN conda create -n realcam python=3.10 -y && \
30
  conda run -n realcam pip install --upgrade pip && \
31
  conda run -n realcam conda install ffmpeg=7 -c conda-forge && \
32
+ conda install pytorch==2.6 torchvision -c pytorch -c nvidia
33
+
34
  conda run -n realcam pip install -r /app/requirements.txt
35
 
36
  RUN chmod -R 777 /app /workspace