美文网首页深度学习
Tensroflow-GPU环境快速搭建

Tensroflow-GPU环境快速搭建

作者: 逆风g | 来源:发表于2018-07-13 13:28 被阅读2次

现在网上的很多Tensorflow-GPU环境安装教程都是这样的:

  1. 安装cuda
  2. 安装cudnn
  3. 安装tensorflow-gpu

本篇文章将教大家直接使用Anaconda来快速搭建自己的tensorlow-gpu版环境。

为什么要用Anaconda?

  • 预先集成了很多数据科学相关的工具包,并且可以使用 conda 来安装、更新 、卸载工具包 。
  • 在conda中可以建立多个虚拟环境,用于隔离不同项目所需的不同版本的工具包,以防止版本上的冲突。比如python2.7环境和python3.6环境。

安装anaconda

  1. 去官网下载anaconda,https://www.anaconda.com/download/
  2. cd到安装包所在目录,安装:bash Anaconda2-5.1.0-Linux-x86_64.sh
  3. 输入conda list,若出现未找到命令,则需要修改环境变量(二选其一):
    永久添加环境变量(影响当前用户)
    vim ~/.bashrc
    永久添加环境变量(影响所有用户)
    vim /etc/profile
    最后一行添加export PATH="/home/ilab-gcf/anaconda2/bin:$PATH"
    记得激活环境变量:
    source ~/.bashrc
    source /etc/profile
  4. 创建虚拟环境:conda create -n your_env_name python=2.7
    虚拟环境相关命令:
    激活source activate your_env_name
    取消source deactivate your_env_name
    删除conda remove -n your_env_name
  5. 由于资源访问速度问题,需要添加国内镜像:
    conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
    conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
    conda config --add channels https://mirrorsss.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
    conda config --set show_channel_urls yes

在已激活的虚拟环境下安装GPU版本的tensorflow

  1. conda上查找tensorflow资源,anaconda search -t conda tensorflow
  2. 找到对应版本的tensorflow-gpu,查看它的安装命令,anaconda show xxx
  3. 依照命令来安装tensorflow-gpu,例如:conda install --channel https://conda.anaconda.org/jj_cio_testing tensorflow-gpu,会默认安装好numpy cuda cudnn tensorflow等依赖库
  4. 测式,输入python命令,再依次输入如下代码:
    import tensorflow as tf  
    hello = tf.constant('Hello, Word!')  
    sess = tf.Session()  
    print(sess.run(hello))  

假若没有报错那么代表环境已经搭建成功!

相关文章

网友评论

    本文标题:Tensroflow-GPU环境快速搭建

    本文链接:https://www.haomeiwen.com/subject/djejpftx.html