Tensorlfow-gpu(1.9) 安装

• 掉进tensorflow-gpu坑的我，终于把坑填上了，今天抽空就把过程写出来，用于分享，也为了备忘。
• 其中最大的问题就是不注意版本！！！！

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

3. CUDA 与 Tensorflow 版本的对应

以下是我配置的环境：

• 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”
• 创建即可

完工！

注：

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