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

Add Pytorch and set sklearn version in Dockerfile (#346)

1.Set scikit-learn==0.20.0 in Dockerfile
2.Update readme.md of dockerile
3.Add PyTorch 0.4.1
4.Add description for 'nnictl stop all'
parent 7508c87d
...@@ -70,8 +70,16 @@ RUN python3 -m pip --no-cache-dir install torch torchvision ...@@ -70,8 +70,16 @@ RUN python3 -m pip --no-cache-dir install torch torchvision
# #
RUN python3 -m pip --no-cache-dir install Keras==2.1.6 RUN python3 -m pip --no-cache-dir install Keras==2.1.6
#sklearn #
RUN python3 -m pip --no-cache-dir install scikit-learn #PyTorch
#
RUN python3 -m pip --no-cache-dir install torch==0.4.1
RUN python3 -m pip install torchvision==0.2.1
#
#sklearn 0.20.0
#
RUN python3 -m pip --no-cache-dir install scikit-learn==0.20.0
ENV PATH=/usr/local/nvidia/bin:/usr/local/cuda/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/root/.local/bin:/usr/bin:/bin:/sbin ENV PATH=/usr/local/nvidia/bin:/usr/local/cuda/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/root/.local/bin:/usr/bin:/bin:/sbin
......
...@@ -9,6 +9,8 @@ numpy 1.14.3,scipy 1.1.0 ...@@ -9,6 +9,8 @@ numpy 1.14.3,scipy 1.1.0
TensorFlow 1.5.0 TensorFlow 1.5.0
PyTorch (Stable) PyTorch (Stable)
Keras 2.1.6 Keras 2.1.6
PyTorch 0.4.1
scikit-learn 0.20.0
NNI v0.3 NNI v0.3
``` ```
You can take this Dockerfile as a reference for your own customized Dockerfile. You can take this Dockerfile as a reference for your own customized Dockerfile.
......
...@@ -73,7 +73,7 @@ def parse_args(): ...@@ -73,7 +73,7 @@ def parse_args():
#parse stop command #parse stop command
parser_stop = subparsers.add_parser('stop', help='stop the experiment') parser_stop = subparsers.add_parser('stop', help='stop the experiment')
parser_stop.add_argument('id', nargs='?', help='the id of experiment, use "all" for all experiments') parser_stop.add_argument('id', nargs='?', help='the id of experiment, use \'all\' to stop all running experiments')
parser_stop.set_defaults(func=stop_experiment) parser_stop.set_defaults(func=stop_experiment)
#parse trial command #parse trial command
......
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