fffiloni commited on
Commit
abbc07f
1 Parent(s): 7b96708

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +9 -4
Dockerfile CHANGED
@@ -56,13 +56,18 @@ RUN pip install --upgrade pip
56
  RUN pip install -r requirements_HF.txt
57
  RUN pip install gradio
58
 
 
 
 
59
  USER root
60
- RUN pip install setuptools
61
- RUN pip install cupy-cuda11x
62
- # Install basicsr (assuming setup.py is in basicsr directory)
63
- RUN python basicsr/setup.py develop
64
 
65
  USER user
 
 
 
66
  # Install additional Python packages
67
  RUN pip install ffmpeg-python
68
  RUN pip install dlib
 
56
  RUN pip install -r requirements_HF.txt
57
  RUN pip install gradio
58
 
59
+ # Install cupy before basicsr setup
60
+ RUN pip install cupy-cuda11x==12.3.0
61
+
62
  USER root
63
+ # Configure git to trust the directory
64
+ RUN git config --global --add safe.directory /home/user/app
65
+ RUN chown -R user:user /home/user/app
 
66
 
67
  USER user
68
+ # Install basicsr
69
+ RUN cd basicsr && python setup.py develop
70
+
71
  # Install additional Python packages
72
  RUN pip install ffmpeg-python
73
  RUN pip install dlib