Skip to content

T
tic
Commit 5194da65 by Yao Wang Committed by Tianqi Chen
Options

Add gluoncv installation (#2464)

parent 02455306
Showing with 6 additions and 2 deletions
docker/Dockerfile.ci_gpu
......@@ -46,6 +46,9 @@ RUN bash /install/ubuntu_install_opengl.sh
COPY install/ubuntu_install_mxnet.sh /install/ubuntu_install_mxnet.sh
RUN bash /install/ubuntu_install_mxnet.sh
COPY install/ubuntu_install_gluoncv.sh /install/ubuntu_install_gluoncv.sh
RUN bash /install/ubuntu_install_gluoncv.sh
COPY install/ubuntu_install_coreml.sh /install/ubuntu_install_coreml.sh
RUN bash /install/ubuntu_install_coreml.sh
......
docker/Dockerfile.demo_cpu
......@@ -21,7 +21,7 @@ RUN echo deb http://apt.llvm.org/xenial/ llvm-toolchain-xenial-6.0 main \
RUN pip3 install matplotlib Image Pillow jupyter[notebook]
# Deep learning frameworks
RUN pip3 install mxnet tensorflow keras
RUN pip3 install mxnet tensorflow keras gluoncv
# Build TVM
COPY install/install_tvm_cpu.sh /install/install_tvm_cpu.sh
......
docker/Dockerfile.demo_gpu
......@@ -21,7 +21,7 @@ RUN echo deb http://apt.llvm.org/xenial/ llvm-toolchain-xenial-6.0 main \
RUN pip3 install matplotlib Image Pillow jupyter[notebook]
# Deep learning frameworks
RUN pip3 install mxnet tensorflow keras
RUN pip3 install mxnet tensorflow keras gluoncv
# Build TVM
COPY install/install_tvm_gpu.sh /install/install_tvm_gpu.sh
......
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