fffiloni commited on
Commit
bf8b103
·
verified ·
1 Parent(s): e048ff5

Update Dockerfile

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