Skip to content

O
openpose

活在明天 / openpose
与 Fork 源项目一致

从无法访问的项目Fork

通知 1
Star 0
Fork 0
O
openpose

前往新版Gitcode，体验更适合开发者的 AI 搜索 >>

提交 2ca40f5c 编写于 作者: J Jacob Mederos 提交者： Gines
浏览文件
操作

CUDA_ARCH is passed to Caffe (#919)

上级 06d0aa7d
隐藏空白更改
内联 并排
Showing with 2 addition and 1 deletion
CMakeLists.txt
浏览文件 @ 2ca40f5c
...@@ -699,7 +699,7 @@ if (UNIX OR APPLE) ...@@ -699,7 +699,7 @@ if (UNIX OR APPLE)
CMAKE_ARGS -DCMAKE_INSTALL_PREFIX:PATH=<INSTALL_DIR> CMAKE_ARGS -DCMAKE_INSTALL_PREFIX:PATH=<INSTALL_DIR>
-DMKLDNN_INSTALL_DIR:PATH=<INSTALL_DIR> -DMKLDNN_INSTALL_DIR:PATH=<INSTALL_DIR>
-DUSE_CUDNN=${USE_CUDNN} -DUSE_CUDNN=${USE_CUDNN}
-DBUILD_python=OFF -DCUDA_ARCH_NAME=${CUDA_ARCH}
-DCPU_ONLY=${CAFFE_CPU_ONLY} -DCPU_ONLY=${CAFFE_CPU_ONLY}
-DUSE_MKL2017_AS_DEFAULT_ENGINE=${CAFFE_CPU_ONLY} -DUSE_MKL2017_AS_DEFAULT_ENGINE=${CAFFE_CPU_ONLY}
-DCMAKE_BUILD_TYPE=Release -DCMAKE_BUILD_TYPE=Release
...@@ -716,6 +716,7 @@ if (UNIX OR APPLE) ...@@ -716,6 +716,7 @@ if (UNIX OR APPLE)
PREFIX ${CAFFE_PREFIX} PREFIX ${CAFFE_PREFIX}
CMAKE_ARGS -DCMAKE_INSTALL_PREFIX:PATH=<INSTALL_DIR> CMAKE_ARGS -DCMAKE_INSTALL_PREFIX:PATH=<INSTALL_DIR>
-DUSE_CUDNN=${USE_CUDNN} -DUSE_CUDNN=${USE_CUDNN}
-DCUDA_ARCH_NAME=${CUDA_ARCH}
-DCPU_ONLY=${CAFFE_CPU_ONLY} -DCPU_ONLY=${CAFFE_CPU_ONLY}
-DCMAKE_BUILD_TYPE=Release -DCMAKE_BUILD_TYPE=Release
-DBUILD_docs=OFF -DBUILD_docs=OFF
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或