Retrigger build
Browse files- Dockerfile +2 -1
Dockerfile
CHANGED
|
@@ -1,4 +1,5 @@
|
|
| 1 |
-
FROM pytorch/pytorch:
|
|
|
|
| 2 |
|
| 3 |
RUN apt-get update && apt-get install -y --no-install-recommends git && \
|
| 4 |
apt-get clean && rm -rf /var/lib/apt/lists/*
|
|
|
|
| 1 |
+
# FROM pytorch/pytorch:latest
|
| 2 |
+
FROM pytorch/pytorch:2.5.1-cuda12.4-cudnn9-devel
|
| 3 |
|
| 4 |
RUN apt-get update && apt-get install -y --no-install-recommends git && \
|
| 5 |
apt-get clean && rm -rf /var/lib/apt/lists/*
|