Quick fix Docker (#363)

Remove "RUN python3 -m pip --no-cache-dir install torch torchvision"

Quick fix Docker (#363)
Remove "RUN python3 -m pip --no-cache-dir install torch torchvision"
183763ef · SparkSnail · GitHub · e3901253 · 183763ef · 183763ef
Unverified Commit 183763ef authored Nov 13, 2018 by SparkSnail Committed by GitHub Nov 13, 2018
Hide whitespace changes
Inline Side-by-side

Showing with 0 additions and 6 deletions

deployment/docker/Dockerfile deployment/docker/Dockerfile +0 -5

deployment/docker/README.md deployment/docker/README.md +0 -1

No files found.
--- a/deployment/docker/Dockerfile
+++ b/deployment/docker/Dockerfile
@@ -60,11 +60,6 @@ RUN python3 -m pip install nni
 #
 RUN python3 -m pip --no-cache-dir install tensorflow-gpu==1.10.0

-#
-#PyTorch
-#
-RUN python3 -m pip --no-cache-dir install torch torchvision
-
 #
 #Keras 2.1.6
 #

--- a/deployment/docker/README.md
+++ b/deployment/docker/README.md
@@ -7,7 +7,6 @@ This is the Dockerfile of nni project. It includes serveral popular deep learnin
 CUDA 9.0, CuDNN 7.0
 numpy 1.14.3,scipy 1.1.0
 TensorFlow 1.5.0
-PyTorch (Stable)
 Keras 2.1.6
 PyTorch 0.4.1
 scikit-learn 0.20.0