fffiloni commited on
Commit
7fbd0ef
·
verified ·
1 Parent(s): 7c034bc

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +4 -1
Dockerfile CHANGED
@@ -56,6 +56,9 @@ RUN pip install --upgrade pip
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
 
@@ -66,7 +69,7 @@ RUN chown -R user:user /home/user/app
66
 
67
  USER user
68
 
69
- # Install basicsr with the --user flag to install in the user's home directory
70
  RUN python basicsr/setup.py develop --user
71
 
72
  # Install additional Python packages
 
56
  RUN pip install -r requirements_HF.txt
57
  RUN pip install gradio
58
 
59
+ # Install specific Cython version known to work with cupy
60
+ RUN pip install 'Cython<3.0'
61
+
62
  # Install cupy before basicsr setup
63
  RUN pip install cupy-cuda11x==12.3.0
64
 
 
69
 
70
  USER user
71
 
72
+ # Install basicsr with the --user flag
73
  RUN python basicsr/setup.py develop --user
74
 
75
  # Install additional Python packages