Spaces:
Runtime error
Runtime error
Updated Dockerfile for GPU Dependencies
Browse files- Dockerfile +3 -0
Dockerfile
CHANGED
@@ -6,6 +6,9 @@ ENV CUDA_HOME=/usr/local/cuda
|
|
6 |
ENV PATH=$CUDA_HOME/bin:$PATH
|
7 |
ENV LD_LIBRARY_PATH=$CUDA_HOME/lib64:$LD_LIBRARY_PATH
|
8 |
|
|
|
|
|
|
|
9 |
# Install NVIDIA driver
|
10 |
RUN apt-get update && apt-get install -y --no-install-recommends nvidia-driver-460
|
11 |
|
|
|
6 |
ENV PATH=$CUDA_HOME/bin:$PATH
|
7 |
ENV LD_LIBRARY_PATH=$CUDA_HOME/lib64:$LD_LIBRARY_PATH
|
8 |
|
9 |
+
# Set DEBIAN_FRONTEND to noninteractive to prevent interactive prompts
|
10 |
+
ENV DEBIAN_FRONTEND=noninteractive
|
11 |
+
|
12 |
# Install NVIDIA driver
|
13 |
RUN apt-get update && apt-get install -y --no-install-recommends nvidia-driver-460
|
14 |
|