Skip to content
Snippets Groups Projects
Unverified Commit 74314d27 authored by Tianqi Chen's avatar Tianqi Chen Committed by GitHub
Browse files

[DOCKER] temporary revert cuda version to cuda8 (#2021)

parent b5b51f0d
No related branches found
No related tags found
No related merge requests found
Hide whitespace changes
Inline Side-by-side
docker/Dockerfile.ci_gpu
+ 1
− 4
View file @ 74314d27
# 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
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment