fffiloni commited on
Commit
d2812b4
1 Parent(s): 322feda

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +3 -3
Dockerfile CHANGED
@@ -1,6 +1,6 @@
1
  # Use the NVIDIA CUDA runtime as a base image
2
- #FROM nvidia/cuda:11.3.1-cudnn8-devel-ubuntu20.04
3
- FROM nvidia/cuda:11.8.0-devel-ubuntu22.04
4
 
5
  # Set environment variables
6
  ENV DEBIAN_FRONTEND=noninteractive
@@ -61,7 +61,7 @@ RUN pip install gradio
61
  RUN pip install 'Cython<3.0'
62
 
63
  # Install cupy using the pre-built wheel for CUDA 11.x
64
- RUN pip install cupy cupy-cuda118
65
 
66
  USER root
67
  # Configure git to trust the directory
 
1
  # Use the NVIDIA CUDA runtime as a base image
2
+ FROM nvidia/cuda:11.3.1-cudnn8-devel-ubuntu20.04
3
+ #FROM nvidia/cuda:11.8.0-devel-ubuntu22.04
4
 
5
  # Set environment variables
6
  ENV DEBIAN_FRONTEND=noninteractive
 
61
  RUN pip install 'Cython<3.0'
62
 
63
  # Install cupy using the pre-built wheel for CUDA 11.x
64
+ RUN pip install cupy
65
 
66
  USER root
67
  # Configure git to trust the directory