电脑配置:
显卡:GTX 1060(6G)
系统:ubuntu14.04
CUDA: cuda-repo-ubuntu1404-8-0-local_8.0.44-1_amd64.deb
cuDNN: cudnn-8.0-linux-x64-v5.1.solitairetheme8
1.安装系统(ubuntu14.04)
1)做系统盘
工具:rufus-3.1.exe;ubuntu-14.04.1-desktop-amd64.iso;一个u盘。
2)安装系统
https://blog.csdn.net/ezhchai/article/details/52891014
安装报错:https://blog.csdn.net/king_haw/article/details/70753324(报错只是1060对应ubuntu14.04才出现,目测其他版本没有问题。)
我是将1060显卡拆下,换上580(只要不是1060)显卡,然后按照教程安装就能成功;安装后进入ubuntu14.04,安装1060显卡驱动,之后关机换上1060显卡即可。
驱动安装:先去NVIDIA官网查看自己电脑具体驱动版本
(https://blog.csdn.net/king_haw/article/details/70143875)
安装双显卡的教程:https://blog.csdn.net/masa_fish/article/details/51882183
2.安装CUDA
安装runfiles格式请参考:https://blog.csdn.net/masa_fish/article/details/51882183
百度搜索CUDA8.0,下载cuda-repo-ubuntu1404-8-0-local_8.0.44-1_amd64.deb(不同时间版本不同,这里以8.0.44-1为例)
打开终端(ctrl+alt+t)输入你下载的CUDA路径,如:
cd /home/myubuntu/download
然后依次输入以下命令(下载界面也有):
sudo dpkg -i cuda-repo-ubuntu1404-8-0-local_8.0.44-1_amd64.deb
sudo apt-get update
sudo apt-get install cuda
设置环境变量:打开.bashrc文件
sudo gedit ~/.bashrc
在文件末尾添加:
export PATH=/usr/local/cuda-8.0/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64:$LD_LIBRARY_PATH
保存然后在终端运行:
source ~/.bashrc
检查:在终端输入 nvcc –V,出现下面结果:
3.安装cuDNN: cudnn-8.0-linux-x64-v5.1.solitairetheme8
百度cudnn下载进入,选择(需要自己注册账号):
进入下载目录
cd /home/…
sudo tar xvf cudnn-8.0-linux-x64-v5.1.solitairetheme8
sudo cp cuda/include/cudnn.h /usr/local/cuda/include/
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/
sudo chmod a+r /usr/local/cuda/include/cudnn.h
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*
4.安装BLAS
sudo apt-get install libatlas-base-dev
5.安装opencv2.4.10(版本根据自己需求)
参考https://blog.csdn.net/oyster9/article/details/77917645
https://blog.csdn.net/Solomon1558/article/details/51967280
将步骤中的这个命令:
cmake -D CMAKE_BUILD_TYPE=RELEASE -D CMAKE_INSTALL_PREFIX=/usr/local ..
改为:
cmake -D CMAKE_BUILD_TYPE=release -D CMAKE_INSTALL_PREFIX=/usr/local -D CUDA_GENERATION=Kepler ..
安装前:
修改graphcuts.cpp文件(搜索即可,不修改会编译报错)
#if !defined (HAVE_CUDA) || defined (CUDA_DISABLER)
改成
#if !defined (HAVE_CUDA) || defined (CUDA_DISABLER) || (CUDART_VERSION >= 8000)
终端输入:
pkg-config --modversion opencv
出现对应版本号即安装成功。
6.安装caffe需要的python包
首先需要安装pip
sudo apt-get install python-pip
安装requirement里面的包
进入caffe的python目录,执行:
for req in $(cat requirements.txt); do pip install $req; done
打开.bashrc文件
sudo gedit ~/.bashrc
在文件末尾添加(添加caffe的python路径):
export PYTHONPATH=/path/to/caffe/python:$PYTHONPATH(需要修改为自己的路径,如)
export PYTHONPATH=/home/myubuntu/download/caffe/python:$PYTHONPATH
保存。
运行source ~/.bashrc
7.编译caffe
在家目录执行
git clone https://github.com/BVLC/caffe.git
安装依赖项:
sudo apt-get install libatlas-base-dev libprotobuf-dev libleveldb-dev libsnappy-dev libopencv-dev libboost-all-dev libhdf5-serial-dev libgflags-dev libgoogle-glog-dev liblmdb-dev protobuf-compiler
打开 caffe 所在目录,找到Makefile.config.example
文件,将其改名为Makefile.config
,打开。将# USE_CUDNN
:
= 1
一行开头的#删除,
在INCLUDE_DIRS :后面添加 /usr/local/cuda-8.0/include
在LIBRARY_DIRS :后面添加 /usr/local/cuda-8.0/lib
将CUDA_DIR := /usr/local/cuda-7.0改为cuda-8.0
保存。
打开终端:
make all
make test
make runtest
make pycaffe
验证:终端下输入
python
import caffe
不报错即成功。
参考教程:http://caffe.berkeleyvision.org/installation.html
https://www.jianshu.com/p/69a10d0a24b9