Tensorlfow-gpu(1.9) 安装

    • 掉进tensorflow-gpu坑的我,终于把坑填上了,今天抽空就把过程写出来,用于分享,也为了备忘。
    • 其中最大的问题就是不注意版本!!!!
    1. 驱动与CUDA版本的对应

    1.png

    1. CUDA与gcc、g++版本的对应

    CUDA9.0只支持 gcc 6.0 以下的版本

    1. CUDA 与 Tensorflow 版本的对应

    2.png

    以下是我配置的环境:

    • Win10
    • CUDA 9.0
    • CUDNN 7.3
    • NVIDIA 398.36
    • Anaconda 3
    • Pycharm Pro

    下面开始进入正题了!

    一、Anaconda3

    • 首先从官网下载安装包并按要求安装
    • 打开“Anaconda Prompt”
    • 更新Anaconda的源
    conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
    conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge 
    conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
    conda config --set show_channel_urls yes
    • 创建 命名为“tensorflow” & python版本为3.6 的虚拟环境
    conda create -n tensorflow python==3.6
    • 激活虚拟环境
    conda activate tensorflow 
    • 安装tensorflow-gpu
    conda install tensorflow-gpu==1.9

    二、 Pycharm Pro

    • 新建工程
    • 选择刚刚创建Anaconda虚拟环境“tensorflow”

    3.png

    • 创建即可
    完工!

    注:

    1. 用Ubuntu安装Tensorflow-gpu时尤其需要注意gcc与g++的版本问题
    2. conda安装tensorflow-gpu时会自动安装 Cuda 和 CuDNN
    3. 按此方法安装只需要注意nvidia的驱动版本
    4. 退出虚拟环境的命令: conda deactivate