FazBrowse GitHub Viewer | Trending |
URL:
| Home
Tools: [Download Repo ZIP]   [Original HTTPS Page]

Add MXNet-GPU and GluonCV & GluonNLP · RTDataScience/docker-python@e93e2fb · GitHub

/ docker-python Public
forked from Kaggle/docker-python

Commit e93e2fb

Browse files
Sergey Sokolov
committed
Add MXNet-GPU and GluonCV & GluonNLP
1 parent 02f8400 commit e93e2fb

2 files changed

Lines changed: 5 additions & 0 deletions

File tree

‎Dockerfile‎

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -141,6 +141,8 @@ RUN apt-get -y install zlib1g-dev liblcms2-dev libwebp-dev libgeos-dev && \
141141
# MXNet
142142
pip install mxnet && \
143143
pip install --upgrade numpy && \
144+
pip install gluonnlp && \
145+
pip install gluoncv && \
144146
# h2o (requires java)
145147
# requires java
146148
apt-get install -y default-jdk && \

‎gpu.Dockerfile‎

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -52,6 +52,9 @@ RUN pip uninstall -y tensorflow && \
5252
conda install -y pytorch torchvision -c pytorch && \
5353
/tmp/clean-layer.sh
5454

55+
RUN pip uninstall -y mxnet &&
56+
pip install mxnet-cu92mkl
57+
5558
# Install GPU-only packages
5659
RUN pip install pycuda && \
5760
pip install cupy-cuda92 && \

0 commit comments

Comments
 (0)
Back | FazBrowse Home | New Git URL