-
Notifications
You must be signed in to change notification settings - Fork 6.8k
[Dependency Update] Dependency update doc #15045
Conversation
@mxnet-label-bot add [Doc, pr-awaiting-review] |
0185d13
to
9fca137
Compare
9fca137
to
40dad43
Compare
@marcoabreu @perdasilva @aaronmarkham for review |
@marcoabreu @perdasilva @aaronmarkham Gentle Ping! |
tools/dependencies/README.md
Outdated
# Take CUDA 10 for example | ||
wget https://developer.nvidia.com/compute/cuda/10.0/Prod/local_installers/cuda_10.0.130_410.48_linux | ||
chmod +x cuda_10.0.130_410.48_linux && sudo ./cuda_10.0.130_410.48_linux | ||
# Installation except: |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
What does this mean?
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Type fixed
tools/dependencies/README.md
Outdated
### CUDA, cuDNN, NCCL | ||
#### 1. Environment Setup | ||
``` | ||
# Take Ubuntu 16.04 for example |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
You could move this out of the code block and introduce what is going on here.
tools/dependencies/README.md
Outdated
``` | ||
# Download the ImageNet on http://image-net.org/download and make record file | ||
# Install prerequisite package | ||
pip2 install psutil --user |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Why pip2?
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
our benchmark script is written in python2 and not compatible with python3
tools/dependencies/README.md
Outdated
|
||
# Install python frontend | ||
cd python | ||
pip3 install -e . --pre |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
maybe just pip? Let users to pick the right version (maybe thorugh viurtualenv)
tools/dependencies/README.md
Outdated
# Wait for 10 - 30 mins, you will find libmxnet.so under the incubator-mxnet/lib | ||
|
||
# Install python frontend | ||
cd python |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
This can be simplified by pip install -e python
tools/dependencies/README.md
Outdated
pip install -e . --pre | ||
# Test MXNet | ||
>>> import mxnet as mx | ||
>>> mx.nd.ones((2, 5) ctx=mx.gpu(0)) |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Maybe you can replace this test using the mxnet.runtime.feature_list()
tools/dependencies/README.md
Outdated
# Clone the testing script | ||
git clone https://github.com/rahul003/deep-learning-benchmark-mirror.git | ||
# command | ||
python2 benchmark_runner.py --framework mxnet --metrics-policy metrics_parameters_images_top_1 --task-name metrics_parameters_images_top_1 --metrics-suffix test --num-gpus 8 --command-to-execute 'python mxnet_benchmark/train_imagenet.py --use-rec --batch-size 128 --dtype float32 --num-data-workers 40 --num-epochs 3 --gpus 0,1,2,3,4,5,6,7 --lr 0.4 --warmup-epochs 5 --last-gamma --mode symbolic --model resnet50_v1b --rec-train /home/ubuntu/data/train-passthrough.rec --rec-train-idx /home/ubuntu/data/train-passthrough.idx --rec-val /home/ubuntu/data/val-passthrough.rec --rec-val-idx /home/ubuntu/data/val-passthrough.idx' --data-set data |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
This benchmark AI flow is outdated and I think it's also not public. Why not just run training script from example directly?
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Also maybe use mnist as an example here. ImageNet is too big.
da90049
to
4c1c654
Compare
4c1c654
to
09f6734
Compare
|
||
The dependencies could be categorized by several groups: BLAS libraries, CPU-based performance boost library, i.e. MKLDNN and GPU-based performance boosting library including CUDA, cuDNN, NCCL. and others including OpenCV, Numpy, S3-related, PS-lite dependencies. The list below shows all the dependencies and their version. Except for CUDA, cuDNN, NCCL which the user is required to install on their environments, we statically link those dependencies into libmxnet.so when we build PyPi package. By doing this, the user can take advantage of these dependencies without being worry about it. | ||
|
||
| Dependencies | MXNet Version | |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Maybe reword this? It's not MXNet version, but version of each package.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
small comment. otherwise LGTM. thanks for your contribution!
@stu1130 Did you want to make any further updates to this? |
@aaronmarkham I think it's good to go |
* first draft * 2nd draft * complete the draft * capitalize the comment * address the comment * address comment
Description
document dependency update
Checklist
Essentials
Please feel free to remove inapplicable items for your PR.
Changes
Comments