From f5d731625b465167f6d6e555317b1c43b3337a5b Mon Sep 17 00:00:00 2001 From: Boris Sekachev <40690378+bsekachev@users.noreply.github.com> Date: Thu, 20 Sep 2018 14:56:09 +0300 Subject: [PATCH] Cuda installation has been fixed (#71) --- cvat/apps/tf_annotation/docker_setup_tf_annotation.sh | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/cvat/apps/tf_annotation/docker_setup_tf_annotation.sh b/cvat/apps/tf_annotation/docker_setup_tf_annotation.sh index 7fe868d0..42dccd54 100755 --- a/cvat/apps/tf_annotation/docker_setup_tf_annotation.sh +++ b/cvat/apps/tf_annotation/docker_setup_tf_annotation.sh @@ -21,7 +21,7 @@ CUDA_PKG_VERSION="9-0=${CUDA_VERSION}-1" echo "export PATH=/usr/local/nvidia/bin:/usr/local/cuda/bin:${PATH}" >> ${HOME}/.bashrc echo "export LD_LIBRARY_PATH=/usr/local/nvidia/lib:/usr/local/nvidia/lib64:${LD_LIBRARY_PATH}" >> ${HOME}/.bashrc -apt-get update && apt-get install -y --no-install-recommends \ +apt-get update && apt-get install -y --no-install-recommends --allow-unauthenticated \ libprotobuf-dev \ libprotoc-dev \ protobuf-compiler \ @@ -31,7 +31,7 @@ apt-get update && apt-get install -y --no-install-recommends \ libcudnn7=$CUDNN_VERSION-1+cuda9.0 && \ ln -s cuda-9.0 /usr/local/cuda && \ rm -rf /var/lib/apt/lists/* \ - /etc/apt/sources.list.d/nvidia-ml.list + /etc/apt/sources.list.d/nvidia-ml.list /etc/apt/sources.list.d/cuda.list pip3 install --no-cache-dir -r "$(cd `dirname $0` && pwd)/requirements.txt"