Skip to content

GitLab

Unverified Commit 183763ef authored by SparkSnail's avatar SparkSnail Committed by GitHub
Browse files

Quick fix Docker (#363) 

Remove "RUN python3 -m pip --no-cache-dir install torch torchvision"
parent e3901253
Hide whitespace changes
Inline Side-by-side
Showing with 0 additions and 6 deletions
deployment/docker/Dockerfile
View file @ 183763ef
...@@ -60,11 +60,6 @@ RUN python3 -m pip install nni ...@@ -60,11 +60,6 @@ RUN python3 -m pip install nni
# #
RUN python3 -m pip --no-cache-dir install tensorflow-gpu==1.10.0 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 #Keras 2.1.6
# #
......
deployment/docker/README.md
View file @ 183763ef
...@@ -7,7 +7,6 @@ This is the Dockerfile of nni project. It includes serveral popular deep learnin ...@@ -7,7 +7,6 @@ This is the Dockerfile of nni project. It includes serveral popular deep learnin
CUDA 9.0, CuDNN 7.0 CUDA 9.0, CuDNN 7.0
numpy 1.14.3,scipy 1.1.0 numpy 1.14.3,scipy 1.1.0
TensorFlow 1.5.0 TensorFlow 1.5.0
PyTorch (Stable)
Keras 2.1.6 Keras 2.1.6
PyTorch 0.4.1 PyTorch 0.4.1
scikit-learn 0.20.0 scikit-learn 0.20.0
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment