【环境】
ubuntu18.04 x64
gtx1080
CUDA Version 9.0.176
cudnn 7402
1,参照链接安装caffe:
$ sudo apt install caffe-cuda
2,更新scikit-image
$ sudo pip install --upgrade scikit-image --user
3,检查是否安装成功
$ python
Python 3.6.7 (default, Oct 22 2018, 11:32:17)
[GCC 8.2.0] on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> import caffe
>>> caffe.__version__
'1.0.0'
>>> print(caffe.__file__)
>>