fffiloni commited on
Commit
c23bea1
·
verified ·
1 Parent(s): e791814

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +1 -1
Dockerfile CHANGED
@@ -60,7 +60,7 @@ RUN pip install gradio
60
  RUN pip install 'Cython<3.0'
61
 
62
  # Install cupy using the pre-built wheel for CUDA 11.x
63
- RUN pip install cupy_cuda11x
64
 
65
  USER root
66
  # Configure git to trust the directory
 
60
  RUN pip install 'Cython<3.0'
61
 
62
  # Install cupy using the pre-built wheel for CUDA 11.x
63
+ RUN pip install cupy-cuda113
64
 
65
  USER root
66
  # Configure git to trust the directory