GTX1080 安装 cuda 8.0 + cuDNN5.1

来源:互联网 发布:ro sql版本 gm账号 编辑:程序博客网 时间:2024/06/06 07:32

ubuntu下按ctrl+alt+f1~f6出现黑屏现象的解决方法
sudo apt-get install bumblebee bumblebee-nvidia primus linux-headers-generic

reboot

sudo service lightdmstop

cuda安装
sudo sh ./cuda_8.0.27_linux.run
(首先要输入: md5sum cuda_8.0.27_linux.run,看看安装文件是否完整。官网上下载 cuda_8.0.27_linux.run 网页有验证码连接)

//进入安装命令行

Do you accept the previously read EULA?

accept/decline/quit: accept

Install NVIDIA Accelerated Graphics Driver for Linux-x86_64****?

(y)es/(n)o/(q)uit: y

Install the CUDA 8.0 Toolkit?

(y)es/(n)o/(q)uit: y

Enter Toolkit Location

Do you want to install a symbolic link at /usr/local/cuda?

(y)es/(n)o/(q)uit: y

Install the CUDA 8.0 Samples?

(y)es/(n)o/(q)uit: y

Enter CUDA Samples Location

设置环境变量

\ sudo gedit /etc/profile
打开/etc/profie文件,在文件末尾添加如下设置:
export PATH=/usr/local/cuda-8.0/bin:$PATH
保存该文件,执行以下命令,使环境变量立即生效。
\$ source /etc/profile

 添加lib路径
$ sudo gedit /etc/ld.so.conf.d/cuda.conf
在/etc/ld.so.conf.d中添加文件cuda.conf,该文件中写入如下内容:
/usr/local/cuda/lib64
保存该文件,执行下列命令使其立即生效。
$ sudo ldconfig

验证安装成功

cd NVIDIA_CUDA-8.0_Samples/1_Utilities/deviceQuery

make

./deviceQuery

//如果显示pass,则代表成功。

配置cuDNN
先下载 cuDNN5.1 8.0的

cd进入cuDNN5.1解压之后的include目录,在命令行进行如下操作:
sudo cp cudnn.h /usr/local/cuda/include/ #复制头文件

再将cd进入lib64目录下的动态文件进行复制和链接:

sudo cp lib* /usr/local/cuda/lib64/ #复制动态链接库

cd /usr/local/cuda/lib64/
sudo rm -rf libcudnn.so libcudnn.so.5 #删除原有动态文件

sudo ln -s libcudnn.so.5.1.5 libcudnn.so.5 #生成软衔接

sudo ln -s libcudnn.so.5 libcudnn.so #生成软链接

PS:根据你下载的cuDNN版本不同,需要对如上命令中的版本进行修改,以上展示的命令是对于cuDNN 5.1.5的。(在lib64 中看 )

编译正确,但是运行./build/tools/caffe.bin time -model examples/mnist/lenet_train_test.prototxt -gpu 0
报错:找不到库

./build/tools/caffe.bin: error while loading shared libraries: libcudnn.so.5: cannot open shared object file: No such file or directory

用ls -al发现是文件权限的问题,因此用下述命令先删除软连接

cd /usr/local/cuda/lib64/
sudo rm -rf libcudnn.so libcudnn.so.5
sudo chmod u=rwx,g=rx,o=rx libcudnn.so.5.1.5
sudo ln -s libcudnn.so.5.1.5 libcudnn.so.5
sudo ln -s libcudnn.so.5 libcudnn.so

0 0
原创粉丝点击