zoukankan      html  css  js  c++  java
  • deepin15.5 安装tensorflowgpu

    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

  • 相关阅读:
    pdf.js使用
    当前时间距离下一个分钟值以0或者5结尾的时间点(单位:毫秒)小于1分钟时,返回下下个以0或5结尾的时间点
    1
    Could not get JDBC Connection; nested exception is java.sql.SQLException: Cannot get a connection, pool error Pool exhausted 异常原因排查
    gcp导出mysql慢日志写入数据库分析
    redis慢日志结构化
    谷歌云监控告警
    google-cloud的sql-rds导出my.cnf
    慢sql 实时钉钉告警
    docker跨网段搭建gp测试环境
  • 原文地址:https://www.cnblogs.com/minglex/p/9464980.html
Copyright © 2011-2022 走看看