点击关注我哦

一篇文章带你避坑安装tensorflow-gpu

大部分时候，如果我们电脑中已经安装了anaconda，便可以使用 conda 进行安装 tensorflow-gpu，还是使用 conda 进行安装，毕竟不再需要配置CUDA、cuDNN这些“神奇”的东西...大部分仅需要一条命令即可。

本篇文章以使用 conda 安装 tensorflow-gpu==1.13.1为例，步骤如下：

1. 使用 conda 命令进行安装：

conda install tensorflow-gpu==1.13.1

大概率会出现类似下列的信息：

不过没关系，是在我们意料之中的，刚才只是试探一下...

所以，我们接下来尝试查看一下目前都存在什么版本

2. 查看现存版本

anaconda search -t conda tensorflow-gpu

运行上述命令后，将会显示类似下列信息：

图片列出了tensorflow的版本信息、适用系统等信息（部分信息截图）

3. 获取安装命令（以上图信息中的mwani/tensorflow-gpu为例）

anaconda show nwani/tensorflow-gpu

运行上述命令，可以获取类似下图信息：

4. 进行安装

conda install --channel https://conda.anaconda.org/nwani tensorflow-gpu

5. 确认安装成功

python


import tensorflow as tf
print(tf.__version__)

若输出下列版本信息，则证明安装成功：

1.13.1

·  END  ·

HAPPY LIFE