|
@@ -31,6 +31,7 @@ ENV CUDA_VERSION 11.1.1
|
|
|
RUN apt-get update && apt-get install -y --no-install-recommends \
|
|
|
cuda-cudart-11-1=11.1.74-1 \
|
|
|
cuda-compat-11-1 \
|
|
|
+ libnvidia-decode-455 \
|
|
|
&& ln -s cuda-11.1 /usr/local/cuda && \
|
|
|
rm -rf /var/lib/apt/lists/*
|
|
|
|
|
@@ -39,7 +40,7 @@ RUN echo "/usr/local/nvidia/lib" >> /etc/ld.so.conf.d/nvidia.conf && \
|
|
|
echo "/usr/local/nvidia/lib64" >> /etc/ld.so.conf.d/nvidia.conf
|
|
|
|
|
|
ENV PATH /usr/local/nvidia/bin:/usr/local/cuda/bin:${PATH}
|
|
|
-ENV LD_LIBRARY_PATH /usr/local/nvidia/lib:/usr/local/nvidia/lib64
|
|
|
+ENV LD_LIBRARY_PATH /usr/local/cuda/lib64
|
|
|
|
|
|
# nvidia-container-runtime
|
|
|
ENV NVIDIA_VISIBLE_DEVICES all
|