소스 검색

Remove unnecessary install; fix default env var value

Gerard Escalante 4 년 전
부모
커밋
29e8aa4020
1개의 변경된 파일1개의 추가작업 그리고 2개의 파일을 삭제
  1. 1 2
      docker/Dockerfile.amd64nvidia

+ 1 - 2
docker/Dockerfile.amd64nvidia

@@ -31,7 +31,6 @@ 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/*
 
@@ -44,5 +43,5 @@ ENV LD_LIBRARY_PATH /usr/local/cuda/lib64
 
 # nvidia-container-runtime
 ENV NVIDIA_VISIBLE_DEVICES all
-ENV NVIDIA_DRIVER_CAPABILITIES compute,utility
+ENV NVIDIA_DRIVER_CAPABILITIES compute,utility,video
 ENV NVIDIA_REQUIRE_CUDA "cuda>=11.1 brand=tesla,driver>=418,driver<419 brand=tesla,driver>=440,driver<441 brand=tesla,driver>=450,driver<451"