Spaces:
Runtime error
Runtime error
Update Dockerfile
Browse files- Dockerfile +1 -1
Dockerfile
CHANGED
@@ -1,5 +1,5 @@
|
|
1 |
|
2 |
-
FROM nvidia/cuda:
|
3 |
ENV DEBIAN_FRONTEND noninteractive
|
4 |
|
5 |
RUN apt-get update -y && apt-get upgrade -y \
|
|
|
1 |
|
2 |
+
FROM nvidia/cuda:12.2.0-base-ubuntu22.04
|
3 |
ENV DEBIAN_FRONTEND noninteractive
|
4 |
|
5 |
RUN apt-get update -y && apt-get upgrade -y \
|