Update Dockerfile
Browse files- Dockerfile +1 -1
Dockerfile
CHANGED
|
@@ -1,4 +1,4 @@
|
|
| 1 |
-
FROM pytorch/pytorch:
|
| 2 |
|
| 3 |
# Set working directory
|
| 4 |
WORKDIR /app
|
|
|
|
| 1 |
+
FROM pytorch/pytorch:2.4.1-cuda11.8-cudnn9-devel
|
| 2 |
|
| 3 |
# Set working directory
|
| 4 |
WORKDIR /app
|