Spaces:
Paused
Paused
Update Dockerfile
Browse files- Dockerfile +3 -2
Dockerfile
CHANGED
|
@@ -1,5 +1,6 @@
|
|
| 1 |
# Use the NVIDIA CUDA runtime as a base image
|
| 2 |
-
FROM nvidia/cuda:11.3.1-cudnn8-devel-ubuntu20.04
|
|
|
|
| 3 |
|
| 4 |
# Set environment variables
|
| 5 |
ENV DEBIAN_FRONTEND=noninteractive
|
|
@@ -60,7 +61,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-
|
| 64 |
|
| 65 |
USER root
|
| 66 |
# 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
|
| 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 cupy-cuda118
|
| 65 |
|
| 66 |
USER root
|
| 67 |
# Configure git to trust the directory
|