联想拯救者(r720)Ubuntu16.04系统下安装cuda8.0+cudnn6.0+opencv+caffe+测试caffe自带MNIST数据集

1. NVIDIA驱动安装的问题
系统设置软件更新中附加驱动会自带nvidia384驱动

  • [ 1]

点击右侧系统设置(点击齿轮标志)
在这里插入图片描述

  • [ 2] 点击系统设置在这里插入图片描述
  • [ 3] 再点击软件更新,最后点击附加驱动。先关闭下面的集成显卡选项,再选第一项的NVIDIA-384专有,会有进度条显示还原,等一会儿,重启电脑。
  • 在这里插入图片描述
  • [ 4] 安装完毕后重启验证是否安装成功
  • nvidia-smi
  • 在这里插入图片描述出现以上类似的输出表示已经安装成功

2. CUDA8.0
CUDA8.0+CUDNN6.0+opencv 3.4.1+caffe下载地址

1 准备好下载文件
把cuda_8.0.61_375.26_linux.run移动到对应目录下(/home 下即可)

在此目录下安装CUDA8.0

sudo ~/cuda_8.0.61_375.26_linux.run
  1. 安装界面中作出如下选择
accept/decline/quit: accept
 
Install NVIDIA Accelerated Graphics Driver for Linux-x86_64 375.26?
(y)es/(n)o/(q)uit: n
 
Install the CUDA 8.0 Toolkit?
(y)es/(n)o/(q)uit: y
 
Enter Toolkit Location
 [ default is /usr/local/cuda-8.0 ]:
 
Do you want to install a symbolic link at /usr/local/cuda?
(y)es/(n)o/(q)uit: y #网上很多教程可能选择n,我这里选择y,方便后面CUDA的Samples安装       #测试成功与否
 
Install the CUDA 8.0 Samples?
(y)es/(n)o/(q)uit: y #网上很多教程可能选择n,我这里选择y,方便后面CUDA的Samples安装       #测试成功与否
  1. 添加环境路径
    找到.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"

[4 ] 然后设置环境变量和动态链接库,在命令行输入:

#在命令行输入:
 1. sudo gedit /etc/profile
#在打开的文件末尾加入:
 2. export PATH=/usr/local/cuda/bin:$PATH
 #保存之后,创建链接文件:
 3. sudo gedit /etc/ld.so.conf.d/cuda.conf
 #在打开的文件中添加如下语句:
 4. /usr/local/cuda/lib64
#然后执行,使立即生效
 5. sudo ldconfig
  1. 测试
cd /usr/local/cuda-8.0/samples/1_Utilities/deviceQuery  
make -j4  
sudo ./deviceQuery 

若看到如下出现则表明CUDA安装成功
在这里插入图片描述

3. CUDNN6.0安装

  • 下载cuDNN,移动到/home目录下安装
    先解压
 tar -zxvf cudnn-8.0-linux-x64-v6.0.tgz
  • 得到一个cuda文件夹,进入之后会有include文件夹和lib64文件夹,在CUDA目录下
    打开终端
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*
1. cd caffe
2. sudo sh data/mnist/get_mnist.sh

下载完之后我们可以看到数据集
在这里插入图片描述

  • [ 2] 将数据转换成LMDB数据类型
    在 caffe目录下,然后输入:
sudo sh examples/mnist/create_mnist.sh

在这里可能遇到问题

Creating lmdb…
./examples/mnist/create_mnist.sh: 17: ./examples/mnist/create_mnist.sh: build/examples
/mnist/convert_mnist_data.bin: not found 

在这里插入图片描述参考链接caffe运行mnist出现convert_mnist_data.bin: not found问题解决

  • [3 ] 训练数据集
sudo ./examples/mnist/train_lenet.sh

训练过程如下
在这里插入图片描述

  • [4 ] 测试训练好的模型
    loss = 0.0281093 (* 1 = 0.0281093 ) 测试为手写1
sh ./build/tools/caffe.bin test -model=examples/mnist/lenet_train_test.prototxt -weights=examples/mnist/lenet_iter_10000.caffemodel -gpu=0 

在这里插入图片描述参考链接caffe深度学习【一】caffe入门,Ubuntu下用LeNet进行MNIST数据集训练
caffe 运行mnist示例
ubuntu16.04的caffe测试mnist出现cannot create cublas handle.cublas won’t be available
Ubuntu14.04+caffe+cuda7.5 环境搭建以及MNIST数据集的训练与测试

欢迎大家看留言,有些问题只供参考,具体问题百度或google
欢迎大家看留言,有些问题只供参考,具体问题百度或google
欢迎大家看留言,有些问题只供参考,具体问题百度或google

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值