Spaces:
Runtime error
Runtime error
7th7th7th7th7th7th7th
Browse files- Dockerfile +1 -1
Dockerfile
CHANGED
@@ -1,5 +1,5 @@
|
|
1 |
# Use an NVIDIA CUDA base image
|
2 |
-
ARG CUDA_IMAGE="12.
|
3 |
FROM nvidia/cuda:${CUDA_IMAGE}
|
4 |
|
5 |
ENV HOST 0.0.0.0
|
|
|
1 |
# Use an NVIDIA CUDA base image
|
2 |
+
ARG CUDA_IMAGE="12.3.1-base-ubuntu22.04"
|
3 |
FROM nvidia/cuda:${CUDA_IMAGE}
|
4 |
|
5 |
ENV HOST 0.0.0.0
|