Add gluoncv installation (#2464)

apache · Jan 19, 2019 · 5194da6 · 5194da6
1 parent 0245530
commit 5194da6
Show file tree

Hide file tree

Showing 4 changed files with 6 additions and 2 deletions.
diff --git a/docker/Dockerfile.ci_gpu b/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
 

diff --git a/docker/Dockerfile.demo_cpu b/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

diff --git a/docker/Dockerfile.demo_gpu b/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

diff --git a/docker/install/ubuntu_install_gluoncv.sh b/docker/install/ubuntu_install_gluoncv.sh
@@ -0,0 +1 @@
+pip3 install gluoncv