Spaces:
Runtime error
Runtime error
Update Dockerfile
Browse files- Dockerfile +4 -0
Dockerfile
CHANGED
|
@@ -1,6 +1,10 @@
|
|
| 1 |
# Use an official PyTorch image with CUDA support as the base image
|
| 2 |
FROM pytorch/pytorch:1.9.0-cuda11.1-cudnn8-devel
|
| 3 |
|
|
|
|
|
|
|
|
|
|
|
|
|
| 4 |
# Install Git and system libraries required for OpenGL without interactive prompts
|
| 5 |
ENV DEBIAN_FRONTEND=noninteractive
|
| 6 |
|
|
|
|
| 1 |
# Use an official PyTorch image with CUDA support as the base image
|
| 2 |
FROM pytorch/pytorch:1.9.0-cuda11.1-cudnn8-devel
|
| 3 |
|
| 4 |
+
# Add NVIDIA CUDA GPG key
|
| 5 |
+
RUN apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/7fa2af80.pub
|
| 6 |
+
|
| 7 |
+
|
| 8 |
# Install Git and system libraries required for OpenGL without interactive prompts
|
| 9 |
ENV DEBIAN_FRONTEND=noninteractive
|
| 10 |
|