Skip to content

T
tic
Commit e8d6d9aa by Siva Committed by Tianqi Chen
Options

[DOCKER] CUDA upgrade to 9.0 to acommodate tensorflow-gpu (1.10.0). (#1761)

parent 1de91d07
Showing with 6 additions and 2 deletions
docker/Dockerfile.ci_gpu
# 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
......
docker/Dockerfile.demo_gpu
# 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
......
Markdown is supported
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