From e8d6d9aa704f49a1d6b6cc72ed277b4af7a9aa56 Mon Sep 17 00:00:00 2001 From: Siva Date: Tue, 25 Sep 2018 08:42:29 +0530 Subject: [PATCH] [DOCKER] CUDA upgrade to 9.0 to acommodate tensorflow-gpu (1.10.0). (#1761) --- docker/Dockerfile.ci_gpu | 5 ++++- docker/Dockerfile.demo_gpu | 2 +- docker/install/ubuntu_install_tensorflow.sh | 1 + 3 files changed, 6 insertions(+), 2 deletions(-) create mode 100644 docker/install/ubuntu_install_tensorflow.sh diff --git a/docker/Dockerfile.ci_gpu b/docker/Dockerfile.ci_gpu index c177ef9d420a..bca16b59366b 100644 --- a/docker/Dockerfile.ci_gpu +++ b/docker/Dockerfile.ci_gpu @@ -1,5 +1,5 @@ # CI docker GPU env -FROM nvidia/cuda:8.0-cudnn7-devel +FROM nvidia/cuda:9.0-cudnn7-devel # Base scripts RUN apt-get update --fix-missing @@ -62,6 +62,9 @@ RUN pip3 install Pillow COPY install/ubuntu_install_vulkan.sh /install/ubuntu_install_vulkan.sh RUN bash /install/ubuntu_install_vulkan.sh +COPY install/ubuntu_install_tensorflow.sh /install/ubuntu_install_tensorflow.sh +RUN bash /install/ubuntu_install_tensorflow.sh + # AutoTVM deps COPY install/ubuntu_install_redis.sh /install/ubuntu_install_redis.sh RUN bash /install/ubuntu_install_redis.sh diff --git a/docker/Dockerfile.demo_gpu b/docker/Dockerfile.demo_gpu index 6f249986e22c..d20293c4ed3d 100644 --- a/docker/Dockerfile.demo_gpu +++ b/docker/Dockerfile.demo_gpu @@ -1,6 +1,6 @@ # Minimum docker image for demo purposes # prebuilt-image: tvmai/demo-gpu -FROM nvidia/cuda:8.0-cudnn7-devel +FROM nvidia/cuda:9.0-cudnn7-devel RUN apt-get update --fix-missing diff --git a/docker/install/ubuntu_install_tensorflow.sh b/docker/install/ubuntu_install_tensorflow.sh new file mode 100644 index 000000000000..407954f8fd46 --- /dev/null +++ b/docker/install/ubuntu_install_tensorflow.sh @@ -0,0 +1 @@ +pip3 install tensorflow-gpu