Update Dockerfile
Browse files- Dockerfile +1 -1
Dockerfile
CHANGED
|
@@ -1,7 +1,7 @@
|
|
| 1 |
ARG CUDA_VERSION=12.6
|
| 2 |
ARG TORCH_BASE=full
|
| 3 |
|
| 4 |
-
FROM pytorch/2.
|
| 5 |
|
| 6 |
LABEL maintainer="XXXXRT"
|
| 7 |
LABEL version="V2 Pro"
|
|
|
|
| 1 |
ARG CUDA_VERSION=12.6
|
| 2 |
ARG TORCH_BASE=full
|
| 3 |
|
| 4 |
+
FROM pytorch/pytorch:2.8.0-cuda12.6-cudnn9-devel
|
| 5 |
|
| 6 |
LABEL maintainer="XXXXRT"
|
| 7 |
LABEL version="V2 Pro"
|