Spaces:
Paused
Paused
Update Dockerfile
Browse files- Dockerfile +1 -1
Dockerfile
CHANGED
|
@@ -1,5 +1,5 @@
|
|
| 1 |
# Base image with PyTorch 2.4.0 + CUDA 12.1
|
| 2 |
-
FROM pytorch/pytorch:2.4.0-cuda12.1-
|
| 3 |
|
| 4 |
# Create a non-root user
|
| 5 |
RUN useradd -m -u 1000 user
|
|
|
|
| 1 |
# Base image with PyTorch 2.4.0 + CUDA 12.1
|
| 2 |
+
FROM pytorch/pytorch:2.4.0-cuda12.1-cudnn9-runtime
|
| 3 |
|
| 4 |
# Create a non-root user
|
| 5 |
RUN useradd -m -u 1000 user
|