[CI] Update PyTorch in CI image to 1.5.1 (#1808)

* test

* Revert "test"

This reverts commit be555b06a9942a3267d173f4be760a04d33be227.

* change

docker/Dockerfile.ci_cpu

@@ -17,6 +17,7 @@ RUN bash /install/ubuntu_install_conda.sh
 ENV CONDA_ALWAYS_YES="true"
 
 COPY install/conda_env/torch_cpu.yml /install/conda_env/torch_cpu.yml
+COPY install/conda_env/torch_cpu_pip.txt /install/conda_env/torch_cpu_pip.txt
 RUN ["/bin/bash", "-i", "-c", "conda env create -f /install/conda_env/torch_cpu.yml"]
 
 COPY install/conda_env/tensorflow_cpu.yml /install/conda_env/tensorflow_cpu.yml

docker/Dockerfile.ci_gpu

@@ -16,6 +16,7 @@ RUN bash /install/ubuntu_install_conda.sh
 ENV CONDA_ALWAYS_YES="true"
 
 COPY install/conda_env/torch_gpu.yml /install/conda_env/torch_gpu.yml
+COPY install/conda_env/torch_gpu_pip.txt /install/conda_env/torch_gpu_pip.txt
 RUN ["/bin/bash", "-i", "-c", "conda env create -f /install/conda_env/torch_gpu.yml"]
 
 COPY install/conda_env/tensorflow_gpu.yml /install/conda_env/tensorflow_gpu.yml

docker/install/conda_env/mxnet_cpu.yml

@@ -3,7 +3,7 @@ dependencies:
   - python=3.6.9
   - pip
   - pip:
-    - mxnet
+    - mxnet==1.6.0
     - pytest
     - nose
     - numpy

docker/install/conda_env/mxnet_gpu.yml

@@ -3,7 +3,7 @@ dependencies:
   - python=3.6.9
   - pip
   - pip:
-    - mxnet-cu101
+    - mxnet-cu101==1.6.0
     - pytest
     - nose
     - numpy

docker/install/conda_env/torch_cpu.yml

@@ -3,15 +3,5 @@ dependencies:
   - python=3.6.9
   - pip
   - pip:
-    - torch
-    - torchvision
-    - pytest
+    - --find-links https://download.pytorch.org/whl/torch_stable.html
+    - --requirement torch_cpu_pip.txt
\ No newline at end of file
-    - nose
-    - numpy
-    - cython
-    - scipy
-    - networkx
-    - matplotlib
-    - nltk
-    - requests[security]
-    - tqdm
\ No newline at end of file

docker/install/conda_env/torch_cpu_pip.txt

+--find-links https://download.pytorch.org/whl/torch_stable.html
+torch==1.5.1+cpu
+torchvision==0.6.1+cpu
+pytest
+nose
+numpy
+cython
+scipy
+networkx
+matplotlib
+nltk
+requests[security]
+tqdm  
\ No newline at end of file

docker/install/conda_env/torch_gpu.yml

@@ -3,15 +3,5 @@ dependencies:
   - python=3.6.9
   - pip
   - pip:
-    - torch
-    - torchvision
-    - pytest
+    - --find-links https://download.pytorch.org/whl/torch_stable.html
+    - --requirement torch_gpu_pip.txt
\ No newline at end of file
-    - nose
-    - numpy
-    - cython
-    - scipy
-    - networkx
-    - matplotlib
-    - nltk
-    - requests[security]
-    - tqdm  
\ No newline at end of file

docker/install/conda_env/torch_gpu_pip.txt

+--find-links https://download.pytorch.org/whl/torch_stable.html
+torch==1.5.1+cu101
+torchvision==0.6.1+cu101
+pytest
+nose
+numpy
+cython
+scipy
+networkx
+matplotlib
+nltk
+requests[security]
+tqdm  
\ No newline at end of file