记录一下安装过程和错误,给自己提醒...
这次装了C3D,因为要用3d卷积核,其实编译过程和普通caffe一样
显卡:NVIDIA GTX 1050Ti
系统:Ubuntu16.04
安装过程参考如下博客,谢谢各位大佬!
ubuntu16.04安装caffe
爆详细Ubuntu16.04,CUDA8.0,OpenCV3.1,Tensorflow完全配置指南
一.显卡驱动
ctrl+alt+t打开终端
添加官方源
$ sudo add-apt-repository ppa:graphics-drivers/ppa
然后去nvidia官网查一下自己的显卡驱动,www.nvidia.cn/Download/index.aspx?lang=cn 我的是384,所以下面输的是nvidia-384
1.$ sudo apt-get update
2.$ sudo apt-get install nvidia-384 nvidia-settings nvidia-prime
安装完之后重启电脑,看是否安装成功
$ nvidia-settings
如果出现如下界面,表示安装成功
二.安装CUDA
去CUDA官网下载run文件https://developer.nvidia.com/cuda-downloads
我的是cuda9.0。下载好之后cd到文件所在位置
sudo ./cuda_9.0.176_384.81_linux.run --override
出现这个表示安装正确
===========
= Summary =
===========
Driver: Not Selected
Toolkit: Installed in /usr/local/cuda-9.0
Samples: Installed in /usr/local/cuda-9.0, but missing recommended librarie
接下来下载cudnn,需要先注册,选library for linux,https://developer.nvidia.com/cudnn
到下载的文件所在的目录解压,我的文件都在home里
1.tar zxvf cudnn-9.0-linux-x64-v7.tgz
2.cd cuda/include/
3.sudo cp cudnn.h /usr/local/cuda/include/ #复制头文件
4.cd ..
5.cd lib64/
6.sudo cp lib* /usr/local/cuda/lib64/ #复制动态链接库
#创建新的链接
7.cd /usr/local/cuda/lib64/
8.sudo rm -rf libcudnn.so libcudnn.so.7
9.sudo ln -s libcudnn.so.7.0.3 libcudnn.so.7
10.sudo ln -s libcudnn.so.7 libcudnn.so
设置环境变量
gedit ~/.bashrc
把下面的路径加进去,=两边不要有空格
export PATH=/usr/local/cuda/bin:$PATH
保存后创建链接文件
1.sudo vim /etc/ld.so.c