使用conda安装。最新的anaconda3自带python版本太高,安装tensorflow-gpu和caffe-gpu问题都太多。
1.先用系统的软件更新 装上显卡驱动。
Ubuntu18.04安装英伟达显卡驱动 - AbelSu - 博客园
2.使用conda命令安装caffe-gpu
简洁Ubuntu16.04 Anaconda安装GPU版caffe
3.链接2中会出现无法安装 protobuf=3.2,因为现在官网下载的anaconda3自带python版本为3.7,从而不支持protobuf3.2。
python降级这一条有可能不需要,因为2中刚开始就是创建了新的python环境。
Anaconda python3.6降级python 3.5 - newlittlewhite的博客 - CSDN博客
4.重新启动终端时,输入
source activate caffe
即切换到caffe python35环境了。大功告成。
CAFFE2安装:(python3.5安装不成功,好像是版本问题,在安装caffe的过程中出现过。)
创建虚拟环境caffe2并激活
一键安装
conda install -c caffe2 caffe2-cuda9.0-cudnn7