您当前的位置:首页 > IT编程 > python
| C语言 | Java | VB | VC | python | Android | TensorFlow | C++ | oracle | 学术与代码 | cnn卷积神经网络 | gnn | 图像修复 | Keras | 数据集 | Neo4j | 自然语言处理 | 深度学习 | 医学CAD | 医学影像 | 超参数 | pointnet | pytorch | 异常检测 | Transformers | 情感分类 | 知识图谱 |

自学教程:Ubuntu + CUDA9.0 + tensorflow-gpu 安装过程

51自学网 2021-11-12 07:37:52
  python
这篇教程Ubuntu + CUDA9.0 + tensorflow-gpu 安装过程写得很实用,希望能帮到您。
https://blog.csdn.net/qq_35976351/article/details/79325476
Ubuntu + CUDA9.0 + tensorflow-gpu 安装过程
简介:

tensorflow支持CUDA9.0和cuDNN7.0,因此本教程是在该版本基础上进行安装的。我的电脑CPU是Intel core i7 4710MQ,GPU是GTX940M,请大家在安装前自己检查下硬件是否符合要求。
Ubuntu 18.04

请更新的系统 https://blog.csdn.net/qq_35976351/article/details/89178917
win10系统

https://blog.csdn.net/qq_35976351/article/details/83041890
系统说明:

最好是Ubuntu16.04的,Ubuntu17.10也可以。CUDA9.0要求GCC版本是5.x或者6.x,其他版本不可以,需要自己进行配置,具体过程在这里不再赘述。其他的系统参照下面的配图,一般来说只要是GCC和硬件满足要求都行:
这里写图片描述
安装CUDA® Toolkit 9.0.

CUDA9.0下载地址:https://developer.nvidia.com/cuda-90-download-archive?target_os=Linux&target_arch=x86_64&target_distro=Ubuntu&target_version=1604&target_type=deblocal

注意,一定是要是CUDA9.0而不是CUDA9.1,否则tensorflow运行时会报错(我第一次就选错了)!!!首次进入下载需要邮件注册,大家找一个邮件注册后就可以了。注册完成后,再进入上边的那个链接就行。
选择版本的过程,按照黑笔画出的标记:
这里写图片描述
假设我们把CUDA的安装包下载到~/Downloads文件夹中。之后,在这里打开终端,依次输入下面的命令:

    sudo dpkg -i cuda-repo-ubuntu1604-9-0-local_9.0.176-1_amd64.deb
    sudo apt-key add /var/cuda-repo-9-0-local/7fa2af80.pub
    sudo apt-get update
    sudo apt-get install cuda

再CUDA完成安装之后,还需要添加环境变量,打开终端,输入下面的命令:

    export PATH=/usr/local/cuda-9.0/bin${PATH:+:${PATH}}
    如果是64位系统,输入:
    export LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
    如果是32位系统,输入:
    export LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

上述过程完成了整个的CUDA9.0的安装
安装cuDNN v7.0.

**cuDNN v7.0.**下载地址:https://developer.nvidia.com/cudnn
用之前下载CUDA注册的邮箱登录就行。

进入之后点击Download,之后进入下载界面,选择上I Agree To the Terms of the cuDNN Software License Agreement的复选框,一次要选择上正确的版本!!!!!看下图黑笔标记的
这里写图片描述

同样的,假设下载目录是~/Downloads,在这里打开终端,之后输入以下命令:
sudo dpkg -i libcudnn7_7.0.5.15-1+cuda9.0_amd64.deb
注意,上述命令中的可能会由于cudnn版本的细微差异而不同,注意tab键补齐就行。之后等待完成cuDNN的安装。
安装Tensorflow-GPU版本

GPU相比于CPU,再运算速度上有很大的优势,这也是本文推荐安装GPU版本的原因。

Tensorflow总共有4中安装方式,在这里,我们使用Google官方推荐的安装方式:Virtualenv,创建一个虚拟Python开发环境。

    安装pip和Virtualenv
    打开终端;
    如果使用Python 2.7的版本,输入sudo apt-get install python-pip python-dev python-virtualenv
    如果使用Python 3.x的版本,输入sudo apt-get install python3-pip python3-dev python-virtualenv
    两者选择一个就行,个人推荐Python 3.x版本

    创造一个虚拟的Python开发环境:
    第一步如果选择Python 2.7版本,终端输入:virtualenv --system-site-packages ~/tensorflow
    第一步如果选择Python 3.x版本,终端输入:virtualenv --system-site-packages -p python3 ~/tensorflow
    注意,~/tensorflow是自己选择的位置并创建的目录。大家可以自行选择其他的位置和命名。本教程创建完成之后,会在用户家目录下看到多出的tensorflow文件夹:
————————————————
版权声明:本文为CSDN博主「Erick_Lv」的原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接及本声明。
原文链接:https://blog.csdn.net/qq_35976351/article/details/79325476
sudo apt update 失败的处理方法
import tensorflow 后出现 google - protobuf 类似错误解决方法
万事OK自学网:51自学网_软件自学网_CAD自学网自学excel、自学PS、自学CAD、自学C语言、自学css3实例,是一个通过网络自主学习工作技能的自学平台,网友喜欢的软件自学网站。