diff --git a/Dockerfile.gpu b/Dockerfile.gpu index 5f462867..6375c9b6 100644 --- a/Dockerfile.gpu +++ b/Dockerfile.gpu @@ -7,6 +7,8 @@ ENV LANG=C.UTF-8 \ LC_ALL=C.UTF-8 \ PATH=$CONDA_DIR/bin:$PATH +RUN apt-key adv --fetch-keys "http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64/3bf863cc.pub" + RUN apt-get update -y && \ apt-get install -y --no-install-recommends \ git \ diff --git a/README.md b/README.md index 8aab285b..fd6e4ad4 100644 --- a/README.md +++ b/README.md @@ -45,7 +45,7 @@ Adjust as appropriate. DISTRO=ubuntu VERSION=1604 ARCH=x86_64 -sudo apt-key adv --fetch-keys http://developer.download.nvidia.com/compute/cuda/repos/${DISTRO}${VERSION}/${ARCH}/7fa2af80.pub +sudo apt-key adv --fetch-keys http://developer.download.nvidia.com/compute/cuda/repos/${DISTRO}${VERSION}/${ARCH}/3bf863cc.pub sudo sh -c 'echo "deb http://developer.download.nvidia.com/compute/cuda/repos/${DISTRO}${VERSION}/${ARCH}/" > /etc/apt/sources.list.d/cuda.list' sudo apt-get update && sudo apt-get install -y --no-install-recommends cuda-drivers ```