Before-Started.md

December 22, 2020 ยท View on GitHub

Below figures show the required dependencies, as well as recommended versions of operating systems, runtime and supported hardware.

Please make sure the lib version or OS version is correct.

  1. Recommended OS: Ubuntu 16.04 or Ubuntu 18.04
  2. Package&Dependency

Note:

  • If you use native system to build NNFusion, we will provide install_dependency_script to install all the dependencies.
  • If you use docker to build NNFusion, we will wrap all the dependencies in the image.
Package/DependencyVerified Version on Ubuntu 18.04Verified Version on Ubuntu 16.04
build-essential12.412.1
CMake3.18.43.18.4
clang3.9.13.9.1
clang-format3.9.13.9.1
git2.17.12.7.4
curl7.587.47.0
zilib1g1.2.111.2.8
zilib1g-dev1.2.111.2.8
libtinfo-dev6.16
unzip66
autoconf2.692.69
automake1.51.11.15
libtool2.4.62.4.6
ca-certificates20190110~18.04.120190110~16.04.1
gcc7.5.05.4.0
g++7.5.05.4.0
gdb8.17.11.1
sqlite33.223.11.0
libsqlite3-dev3.223.11.0
libcurl4-openssl-dev7.58.07.47.0
libprotobuf-dev3.6.13.6.1
protobuf-compiler3.6.13.6.1
libgflags-dev2.2.12.1.2
libgtest-dev1.9.01.9.0
  1. Runtime
RuntimeVerified Version on Ubuntu 18.04Verified Version on Ubuntu 16.04
CUDA10NA
ROCM3.5NA
  1. The recommended Tensorflow version is 1.14 and below.
  2. Supported GPU cards
Supported GPU cards
Nvidia V100
Nvidia P100
Nvidia 1080
Nvidia 1080ti
Nvidia 2080ti
AMD Vega 20

After you check all the dependencies, versions of OS and etc. You are ready to use NNFusion to compile your model. NNFusion support models of pb and ONNX format. Please refer to Freeze TensorFlow Model and Freeze PyTorch Model.

Refer to Build Guide to learn how to compile and install NNFusion.