Skip to content

Commit 91ccacd

Browse files
committed
Add cuDNN for inference neural network
1 parent 1b5b36a commit 91ccacd

2 files changed

Lines changed: 17 additions & 3 deletions

File tree

README.md

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,7 @@ Nebula uses g++ and nvcc to compile C++ codes to execute on CPUs and NVIDIA GPUs
1818
* nvcc-9.0 or later
1919
* OpenCV-3.2 or later
2020
* OpenBLAS-0.2 or later: optional for CPU acceleration
21-
* cuBLAS and cuDNN (of package nvidia-384 or later): optional for GPU acceleration
21+
* cuBLAS (of package nvidia-384 or later) and cuDNN (version 7.0.4 to 7.6.5): optional for GPU acceleration
2222

2323
In Ubuntu 18.04, use the following command to install the required libraries except for the NVIDIA driver package.
2424

@@ -28,6 +28,13 @@ To install an NVIDIA driver package for cuBLAS and cuDNN libraries, refer to the
2828

2929
$ sudo ./cuda_9.0.176_384.81_linux-run
3030

31+
To install cuDNN libraries, refer to the following link:
32+
https://docs.nvidia.com/deeplearning/cudnn/install-guide/index.html. For example, installing cudnn-v7.6.5 can be done by executing the following commands with sudo privilege.
33+
34+
$ tar xf cudnn-10.2-linux-x64-v7.6.5.32.tgz
35+
$ sudo cp cuda/include/cudnn*.h /usr/local/cuda/include
36+
$ sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
37+
$ sudo chmod a+r /usr/local/cuda/include/cudnn*.h /usr/local/cuda/lib64/libcudnn*
3138

3239
## Download
3340
The latest release of Nebula benchmark suite is v1.1 (as of Aug., 2020). To obatin a copy of Nebula v1.1, use the following git command in a terminal.
@@ -38,7 +45,6 @@ Or, if you wish to use the latest development version, simply clone the git resp
3845

3946
$ git clone https://github.com/yonsei-icsl/nebula
4047

41-
4248
## Build
4349
Nebula provides a script file named nebula.sh to facilitate the build and run processes of benchmark suite. To build the entire models of Nebula suite, execute the script file as follows in the main directory of Nebula.
4450

UPDATE

Lines changed: 9 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -8,6 +8,14 @@
88
#######################################################
99

1010
Update history:
11-
* Version 1.0 (July 09, 2020)
11+
* Version 1.0 (July, 2020)
1212
- Initial release
1313
- Support convolutional and fully-connected networks.
14+
15+
* Version 1.1 (Aug, 2020)
16+
- Fix some bugs
17+
- Add namespace named nebula
18+
- Update Documentation
19+
20+
* Version 1.2 (Aug, 2020)
21+
- Support cuDNN libraries for faster inference of nebula benchmarks.

0 commit comments

Comments
 (0)