update docker (dmlc#1571)

PawelGlomski-Intel · May 18, 2021 · df52171 · df52171
1 parent 5122e84
commit df52171
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 7 deletions.
diff --git a/tools/batch/submit-job.py b/tools/batch/submit-job.py
@@ -10,7 +10,7 @@
 
 instance_type_info = {
     'g4dn.4x': {
-        'job_definition': 'gluon-nlp-g4dn_4xlarge:5',
+        'job_definition': 'gluon-nlp-g4dn_4xlarge:6',
         'job_queue': 'g4dn'
     },
     'g4dn.8x': {

diff --git a/tools/docker/gluon_nlp_job.sh b/tools/docker/gluon_nlp_job.sh
@@ -23,14 +23,10 @@ git fetch origin $SOURCE_REF:working
 git checkout working
 
 if [ $DEVICE == "cpu" ]; then
+  python3 -m pip uninstall --quiet mxnet -y
   python3 -m pip install -U --quiet --pre "mxnet>=2.0.0b20210121" --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 uninstall --quiet mxnet-cu102 -y
   python3 -m pip install -U --quiet --pre "mxnet-cu102>=2.0.0a" --user
 fi