Skip to content

Commit

Permalink
[Docker] Update Docker (dmlc#1454)
Browse files Browse the repository at this point in the history
* update docker

* update docker

* update readme

* update workflows
  • Loading branch information
barry-jin authored Dec 8, 2020
1 parent 33b6218 commit d4a1d5d
Show file tree
Hide file tree
Showing 6 changed files with 11 additions and 11 deletions.
2 changes: 1 addition & 1 deletion .github/workflows/buildwebsite.yml
Original file line number Diff line number Diff line change
Expand Up @@ -33,7 +33,7 @@ jobs:
run: |
python -m pip install --quiet --upgrade pip
python -m pip install --upgrade cython
python -m pip install --pre "mxnet>=2.0.0b20201127" -f https://dist.mxnet.io/python
python -m pip install --pre "mxnet>=2.0.0b20201206" -f https://dist.mxnet.io/python
python -m pip install setuptools ipython sphinx>=1.5.5 sphinx-gallery nbsphinx sphinx_rtd_theme mxtheme sphinx-autodoc-typehints matplotlib Image recommonmark
- name: Checkout repository(For push)
Expand Down
2 changes: 1 addition & 1 deletion .github/workflows/unittests.yml
Original file line number Diff line number Diff line change
Expand Up @@ -55,7 +55,7 @@ jobs:
python -m pip install --upgrade pip
python -m pip install setuptools pytest pytest-cov contextvars
python -m pip install --upgrade cython
python -m pip install --pre "mxnet>=2.0.0b20200802" -f https://dist.mxnet.io/python
python -m pip install --pre "mxnet>=2.0.0b20201206" -f https://dist.mxnet.io/python
python -m pip install -U -e .[extras]
- name: Build and Install TVM
if: matrix.os == 'ubuntu-latest'
Expand Down
10 changes: 5 additions & 5 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -28,19 +28,19 @@ First of all, install the latest MXNet. You may use the following commands:

```bash
# Install the version with CUDA 10.0
python3 -m pip install -U --pre "mxnet-cu100>=2.0.0b20201121" -f https://dist.mxnet.io/python
python3 -m pip install -U --pre "mxnet-cu100>=2.0.0b20201206" -f https://dist.mxnet.io/python

# Install the version with CUDA 10.1
python3 -m pip install -U --pre "mxnet-cu101>=2.0.0b20201121" -f https://dist.mxnet.io/python
python3 -m pip install -U --pre "mxnet-cu101>=2.0.0b20201206" -f https://dist.mxnet.io/python

# Install the version with CUDA 10.2
python3 -m pip install -U --pre "mxnet-cu102>=2.0.0b20201121" -f https://dist.mxnet.io/python
python3 -m pip install -U --pre "mxnet-cu102>=2.0.0b20201206" -f https://dist.mxnet.io/python

# Install the version with CUDA 11
python3 -m pip install -U --pre "mxnet-cu110>=2.0.0b20201121" -f https://dist.mxnet.io/python
python3 -m pip install -U --pre "mxnet-cu110>=2.0.0b20201206" -f https://dist.mxnet.io/python

# Install the cpu-only version
python3 -m pip install -U --pre "mxnet>=2.0.0b20201121" -f https://dist.mxnet.io/python
python3 -m pip install -U --pre "mxnet>=2.0.0b20201206" -f https://dist.mxnet.io/python
```


Expand Down
4 changes: 2 additions & 2 deletions tools/docker/gluon_nlp_job.sh
Original file line number Diff line number Diff line change
Expand Up @@ -23,15 +23,15 @@ git fetch origin $SOURCE_REF:working
git checkout working

if [ $DEVICE == "cpu" ]; then
python3 -m pip install -U --quiet --pre "mxnet>=2.0.0b20201121" -f https://dist.mxnet.io/python --user
python3 -m pip install -U --quiet --pre "mxnet>=2.0.0b20201206" -f https://dist.mxnet.io/python --user
else
# Due to the issue in https://forums.aws.amazon.com/thread.jspa?messageID=953912
# We need to manually configure the shm to ensure that Horovod is runnable.
# The reason that we need a larger shm is described in https://github.com/NVIDIA/nccl/issues/290
umount shm
mount -t tmpfs -o rw,nosuid,nodev,noexec,relatime,size=2G shm /dev/shm
sudo python3 -m pip uninstall --quiet mxnet-cu102
python3 -m pip install -U --quiet --pre "mxnet-cu102>=2.0.0b20201121" -f https://dist.mxnet.io/python/cu102 --user
python3 -m pip install -U --quiet --pre "mxnet-cu102>=2.0.0b20201206" -f https://dist.mxnet.io/python/cu102 --user
fi

python3 -m pip install --quiet -e .[extras]
Expand Down
2 changes: 1 addition & 1 deletion tools/docker/ubuntu18.04-cpu.Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -33,7 +33,7 @@ RUN bash /install/install_python_packages.sh
RUN bash /install/install_tvm_cpu.sh

# Install MXNet
RUN python3 -m pip install -U --pre "mxnet>=2.0.0b20201121" -f https://dist.mxnet.io/python --user
RUN python3 -m pip install -U --pre "mxnet>=2.0.0b20201206" -f https://dist.mxnet.io/python --user

# Install PyTorch
RUN python3 -m pip install -U torch torchvision --user
Expand Down
2 changes: 1 addition & 1 deletion tools/docker/ubuntu18.04-gpu.Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -32,7 +32,7 @@ RUN bash /install/install_python_packages.sh
RUN bash /install/install_tvm_gpu.sh

# Install MXNet
RUN python3 -m pip install -U --pre "mxnet-cu102>=2.0.0b20201121" -f https://dist.mxnet.io/python --user
RUN python3 -m pip install -U --pre "mxnet-cu102>=2.0.0b20201206" -f https://dist.mxnet.io/python --user

# Install PyTorch
RUN python3 -m pip install -U torch torchvision --user
Expand Down

0 comments on commit d4a1d5d

Please sign in to comment.