diff options
| author | Evan Shelhamer <shelhamer@imaginarynumber.net> | 2017-01-17 09:12:01 -0800 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2017-01-17 09:12:01 -0800 |
| commit | 536851c00e44d545649648b372503f52a89c0499 (patch) | |
| tree | d8f841db4fec3a92bf2cbe9a651b2bdf2d29159f /CMakeLists.txt | |
| parent | 0a91794d85562eaa6064541a58a3588b5afb0e5f (diff) | |
| parent | 5f28eb1147c1abb6e5e5c7cd282218679b0d531d (diff) | |
| download | caffeonacl-536851c00e44d545649648b372503f52a89c0499.tar.gz caffeonacl-536851c00e44d545649648b372503f52a89c0499.tar.bz2 caffeonacl-536851c00e44d545649648b372503f52a89c0499.zip | |
Merge pull request #4563 from cypof/nccl
adopt NVIDIA's NCCL for multi-GPU and switch interface to python
Diffstat (limited to 'CMakeLists.txt')
| -rw-r--r-- | CMakeLists.txt | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/CMakeLists.txt b/CMakeLists.txt index da7142c9..3af394f7 100644 --- a/CMakeLists.txt +++ b/CMakeLists.txt @@ -28,6 +28,7 @@ include(cmake/ConfigGen.cmake) # ---[ Options caffe_option(CPU_ONLY "Build Caffe without CUDA support" OFF) # TODO: rename to USE_CUDA caffe_option(USE_CUDNN "Build Caffe with cuDNN library support" ON IF NOT CPU_ONLY) +caffe_option(USE_NCCL "Build Caffe with NCCL library support" OFF) caffe_option(BUILD_SHARED_LIBS "Build shared libraries" ON) caffe_option(BUILD_python "Build Python wrapper" ON) set(python_version "2" CACHE STRING "Specify which Python version to use") |