Ubuntu18.04+cuda 10.0 配置caffe-action(TSN)
由于使用TSN时,需要配置指定的caffe,网上查阅许多都是说支持cuda8.0,尴尬了!!cuda 10.0怎么破,容我试试,通过解决一些问题还真成功了。
配置
- ubuntu:18.04;
- cuda:10.0。
// An highlighted block
var foo = 'bar';
执行过程及问题解决方法
1. 下载caffe-action。下载地址;
2. 修改Makefile.config,Makefile保持不变;
#cuDNN acceleration switch (uncomment to build with cuDNN).
USE_CUDNN := 1
…
WITH_PYTHON_LAYER := 1
#Whatever else you find you need goes here.
INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include /usr/include/hdf5/serial
LIBRARY_DIRS := $(PYTHON_LIB) /usr/local/lib /usr/lib /usr/lib/x86_64-linux-gnu /usr/lib/x86_64-linux-gnu/hdf5/se