我是按照GitHub上的说明安装的,步骤如下
- git clone [email protected]:chuckcho/video-caffe.git
- cd video-caffe
- mkdir build && cd build
- cmake ..
Make sure CUDA and CuDNN are detected and their paths are correct. - make all
- make install
- (optional) make runtest
我的显卡不支持pascal 模式,虽然修改了makefile.config
但是依然出现了 compute_60的错误。
所以我修改了caffe/cmake/Cuda.cmake
将Caffe_known_gpu_archs 后面的60 61去掉
#set(Caffe_known_gpu_archs "20 21(20) 30 35 50 60 61")
set(Caffe_known_gpu_archs "20 21(20) 30 35 50")
同时注释掉了Pascal的部分
if(${CUDA_ARCH_NAME} STREQUAL "Fermi")
set(__cuda_arch_bin "20 21(20)")
elseif(${CUDA_ARCH_NAME} STREQUAL "Kepler")
set(__cuda_arch_bin "30 35")
elseif(${CUDA_ARCH_NAME} STREQUAL "Maxwell")
set(__cuda_arch_bin "50")
# elseif(${CUDA_ARCH_NAME} STREQUAL "Pascal")
# set(__cuda_arch_bin "60 61")
elseif(${CUDA_ARCH_NAME} STREQUAL "All")
set(__cuda_arch_bin ${Caffe_known_gpu_archs})
elseif(${CUDA_ARCH_NAME} STREQUAL "Auto")
caffe_detect_installed_gpus(__cuda_arch_bin)
else() # (${CUDA_ARCH_NAME} STREQUAL "Manual")
set(__cuda_arch_bin ${CUDA_ARCH_BIN})
endif()
完成这两处修改,运行成功。