saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-30 20:18:08 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Update BUILD.md training dependency info. (#5240)

Browse source 
Update training dependency versions based on Dockerfile.training.
This commit is contained in:
edgchen1 2020-09-22 10:36:04 -07:00 committed by GitHub
parent d957dbebea
commit 14f250a4d0
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 7 additions and 13 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

20
BUILD.md
View file

@ -1109,19 +1109,13 @@ Dockerfile instructions are available [here](./dockerfiles#migraphx)
The default NVIDIA GPU build requires CUDA runtime libraries installed on the system:
* CUDA 10.1
* cuDNN 7.6.2
* NCCL v2.4.8 (download v2.4.8 from the Legacy downloads page)
* OpenMPI 4.0.0.0
```
wget https://download.open-mpi.org/release/open-mpi/v4.0/openmpi-4.0.0.tar.gz
tar zxf openmpi-4.0.0.tar.gz
cd openmpi-4.0.0
./configure --enable-orterun-prefix-by-default
make -j $(nproc) all
sudo make install
sudo ldconfig
```
* CUDA 10.2
* cuDNN 7.6.5
* NCCL v2.7.8
* OpenMPI 4.0.4
The official dependency versions are specified in [Dockerfile.training](./dockerfiles/Dockerfile.training).
## Build instructions
1. Checkout this code repo with `git clone https://github.com/microsoft/onnxruntime`