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

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 https://github.com/cupy/cupy/releases/download/v12.3.0/cupy_cuda11x-12.3.0-cp38-cp38-manylinux2014_aarch64.whl
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_cuda11x
64
 
65
  USER root
66
  # Configure git to trust the directory