fffiloni commited on
Commit
e57feba
·
verified ·
1 Parent(s): 18e7b3c

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-cuda11x==13.3.0
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-cuda113==10.6.0
56
 
57
  USER root
58
  # Install basicsr (assuming setup.py is in basicsr directory)