Skip to content

T
tic
Unverified Commit 9e9cda54 by Tianqi Chen Committed by GitHub
Options

[DOCKER] add ci tag, upgrade gpu-ci (#2438)

parent f7beea4b
Showing with 5 additions and 1 deletions
docker/Dockerfile.ci_gpu
# CI docker GPU env # CI docker GPU env
# tag: v0.41
FROM nvidia/cuda:8.0-cudnn7-devel FROM nvidia/cuda:8.0-cudnn7-devel
# Base scripts # Base scripts
...@@ -48,6 +49,9 @@ RUN bash /install/ubuntu_install_mxnet.sh ...@@ -48,6 +49,9 @@ RUN bash /install/ubuntu_install_mxnet.sh
COPY install/ubuntu_install_coreml.sh /install/ubuntu_install_coreml.sh COPY install/ubuntu_install_coreml.sh /install/ubuntu_install_coreml.sh
RUN bash /install/ubuntu_install_coreml.sh RUN bash /install/ubuntu_install_coreml.sh
COPY install/ubuntu_install_tensorflow.sh /install/ubuntu_install_tensorflow.sh
RUN bash /install/ubuntu_install_tensorflow.sh
COPY install/ubuntu_install_keras.sh /install/ubuntu_install_keras.sh COPY install/ubuntu_install_keras.sh /install/ubuntu_install_keras.sh
RUN bash /install/ubuntu_install_keras.sh RUN bash /install/ubuntu_install_keras.sh
......
docker/install/ubuntu_install_tensorflow.sh
pip3 install tensorflow-gpu pip3 install tensorflow
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