diff --git a/Dockerfile.gpu b/Dockerfile.gpu index 19c2d027..5a0958ef 100644 --- a/Dockerfile.gpu +++ b/Dockerfile.gpu @@ -1,5 +1,7 @@ FROM tensorflow/tensorflow:2.0.0-gpu-py3 +RUN apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/3bf863cc.pub + # Install system packages RUN apt-get update && apt-get install -y --no-install-recommends \ bzip2 \