Update

DLA-Ranker.yml

+name: null
+
+channels:
+  - pytorch
+  - conda-forge
+  - defaults
+
+dependencies:
+  - bokeh=1.4
+  - cmake=3.16 # insures that Gloo library extensions will be built
+  - cudnn=7.6
+  - cupti=10.1
+  - cxx-compiler=1.0 # insures C and C++ compilers are available
+  - jupyterlab=1.2
+  - mpi4py=3.0 # installs cuda-aware openmpi
+  - nccl=2.5
+  - nodejs=13
+  - nvcc_linux-64=10.1 # configures environment to be "cuda-aware"
+  - pip=20.0
+  - pip:
+    - mxnet-cu101mkl==1.6.* # MXNET is installed prior to horovod
+    - -r file:requirements.txt
+  - python=3.7
+  - pytorch=1.4
+  - tensorboard=2.2
+  - tensorflow-gpu=2.2
+  - torchvision=0.5