1. 程式人生 > >caffe通過CMakeLists.txt編譯適用於自定義GPU ARCH的動態庫

caffe通過CMakeLists.txt編譯適用於自定義GPU ARCH的動態庫

1.修改caffe目錄下cmake資料夾下Cuda.cmake檔案,修改下面這一行:

set(Caffe_known_gpu_archs "30 35 50 52 61")

2.編譯的時候設定編譯選項:

cmake -DCUDA_ARCH_NAME=All ..