From e3bebff5c50ba260f9e3175acf37b3984131e632 Mon Sep 17 00:00:00 2001 From: tqchen Date: Sat, 27 Oct 2018 20:56:36 -0700 Subject: [PATCH] [DOCKER] temporary revert cuda version to cuda8 --- docker/Dockerfile.ci_gpu | 5 +---- 1 file changed, 1 insertion(+), 4 deletions(-) diff --git a/docker/Dockerfile.ci_gpu b/docker/Dockerfile.ci_gpu index 7b97a54185f4..708331d3d61a 100644 --- a/docker/Dockerfile.ci_gpu +++ b/docker/Dockerfile.ci_gpu @@ -1,5 +1,5 @@ # CI docker GPU env -FROM nvidia/cuda:9.0-cudnn7-devel +FROM nvidia/cuda:8.0-cudnn7-devel # Base scripts RUN apt-get update --fix-missing @@ -62,9 +62,6 @@ 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