Update Dockerfile
Browse files- Dockerfile +2 -2
Dockerfile
CHANGED
|
@@ -1,5 +1,5 @@
|
|
| 1 |
-
# Use
|
| 2 |
-
FROM pytorch/pytorch:2.0.1-cuda11.
|
| 3 |
|
| 4 |
# Install Git and system libraries required for OpenGL without interactive prompts
|
| 5 |
ENV DEBIAN_FRONTEND=noninteractive
|
|
|
|
| 1 |
+
# Use an official PyTorch image with CUDA support as the base image
|
| 2 |
+
FROM pytorch/pytorch:2.0.1-cuda11.7-cudnn8-runtime
|
| 3 |
|
| 4 |
# Install Git and system libraries required for OpenGL without interactive prompts
|
| 5 |
ENV DEBIAN_FRONTEND=noninteractive
|