fffiloni commited on
Commit
9aea3c7
1 Parent(s): 1bfdf22

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +1 -1
Dockerfile CHANGED
@@ -52,7 +52,7 @@ COPY requirements_HF.txt .
52
  RUN pip install --upgrade pip
53
  RUN pip install -r requirements_HF.txt
54
  RUN pip install gradio
55
- RUN pip install cupy-wheel
56
 
57
  USER root
58
  # Install basicsr (assuming setup.py is in basicsr directory)
 
52
  RUN pip install --upgrade pip
53
  RUN pip install -r requirements_HF.txt
54
  RUN pip install gradio
55
+ RUN pip install cupy==10.4.0
56
 
57
  USER root
58
  # Install basicsr (assuming setup.py is in basicsr directory)