Spaces:
Sleeping
Sleeping
Update Dockerfile
Browse files- Dockerfile +1 -5
Dockerfile
CHANGED
|
@@ -1,8 +1,4 @@
|
|
| 1 |
-
|
| 2 |
-
ARG CUDA="11.1"
|
| 3 |
-
ARG CUDNN="8"
|
| 4 |
-
|
| 5 |
-
FROM pytorch/pytorch:${PYTORCH}-cuda${CUDA}-cudnn${CUDNN}-devel
|
| 6 |
|
| 7 |
ARG DEBIAN_FRONTEND=noninteractive
|
| 8 |
ENV PYTHONUNBUFFERED=1
|
|
|
|
| 1 |
+
FROM nvidia/cuda:11.8.0-base-ubuntu22.04
|
|
|
|
|
|
|
|
|
|
|
|
|
| 2 |
|
| 3 |
ARG DEBIAN_FRONTEND=noninteractive
|
| 4 |
ENV PYTHONUNBUFFERED=1
|