deepin的CUDA和cuDNN安装方法与其它系统有所不同,参考其它操作系统的方法也许不适用,特别是显卡驱动的安装,容易使系统出现问题

本次配置:

操作系统:deepin15.5桌面版

电脑品牌:联想拯救者

CPU:i5-7300HQ

显卡:NVIDIA  GTX 1050Ti

tensorflow版本:1.4

cuda版本:8.0

cudnn版本:6

目录:

  • 安装显卡驱动
  • 安装CUDA
  • 安装tensorflow-GPU
  • 安装CUDNN

一、安装显卡驱动

deepin15.5的显卡驱动可以使用bumblebee工具,此处不需要手动禁用nouveau,因为安装完成之后会自动禁用

安装驱动命令:

sudo apt-get install bumblebee bumblebee-nvidia nvidia-smi

安装完之后,在命令行输入:

optirun nvidia-smi

输出:

上图表明安装已经成功。

驱动使用方法:

在进行训练前,在终端输入sudo tee /proc/acpi/bbswitch <<< ON,可开启显卡。 
在训练结束后,在终端输入sudo tee /proc/acpi/bbswitch <<< OFF,可关闭显卡。

二、安装CUDA

网上很多教程都不适用于我的系统,下载安装包来安装也会报错,所以这里采用命令行安装的方法:

sudo apt install nvidia-cuda-dev nvidia-cuda-toolkit nvidia-nsight nvidia-visual-profiler

至此,CUDA安装完成

三、安装tensorflow-GPU

为什么要在安装cuDNN之前安装tensorflow-gpu呢?因为我们之前安装的cuda方法与往常不同,所以cuDNN的安装方法也有所不同,需要先安装tensorflow-gpu

安装很简单,输入以下命令:

pip install tensorflow-gpu==1.4

四、安装cuDNN

从官方网站下载安装包:https://developer.nvidia.com/rdp/cudnn-download

下载之后解压,会得到一个cuda文件夹。

接下来进行两个步骤:

  • 步骤一:将解压出来的cuda/include/下的cudnn.h文件复制到/usr/local/include/目录下。
  • 步骤二:将cuda/lib64/目录下的所有文件复制到python环境的tensorflow包的tensorflow/python/目录下

我的python环境的tensorflow包的tensorflow/python/目录是:/home/XXX/anaconda3/lib/python3.6/site-packages/tensorflow/python

至此,所有安装都已经结束

测试:

进入python环境中,输入以下命令:

import tensorflow as tf
sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))

输出:

结果中有GPU的名称,显存等信息,表示tensorflow可以使用GPU了

如有问题,欢迎留言讨论

参考资料:

https://blog.csdn.net/aaronjny/article/details/79330727

最新文章

  1. RavenDB官网文档翻译系列第一
  2. Nova PhoneGap框架 第四章 本地数据库
  3. 【codevs1170】 双栈排序
  4. BizTalk开发系列(二十四) BizTalk项目框架建议
  5. 161220、使用Spring AOP实现MySQL数据库读写分离案例分析
  6. 在silverlight中通过WCF连接ORACLE DB数据库(转)
  7. 阿里商业评论 | 互联网POI数据及其在营销中的应用
  8. 数往知来C#面向对象准备〈二〉
  9. python(1) - 列表和元组
  10. Linq保留字含义
  11. HTMLElement
  12. C# 执行oracle sql 语句出现中文不兼容的问题
  13. DatakeyNames和datakey
  14. python subprocess中ssh命令的特殊性
  15. IE, Firefox下,checkbox的钩钩一旦勾上,画面再刷新,钩钩还是勾上的解决方案
  16. 凌华AMP-204C卡的CNC功能 即Feeder工具的使用。
  17. Android布局属性说明
  18. 自动化测试入门指南(3)-- 入门demo
  19. ssm框架基础搭建
  20. Emmet插件使用

热门文章

  1. mybatis批量插入应用
  2. 使用.NET Core中创建Windows服务(一) - 使用官方推荐方式
  3. java基础之和String相关的一些转换
  4. 使用XAMPP配置Apache服务器反向代理
  5. 二 mysql库表的详细操作
  6. Flume 知识点(六)Flume 的监控
  7. 基于SpringBoot+WebSocket搭建一个简单的多人聊天系统
  8. 彻底理解CORS跨域原理
  9. Docker 为非root用户授权
  10. php导出excel乱码怎么处理