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
|