在win10中利用Anaconda直接安装tensorflow-gpu 不需要另行安装cuda cudnn

2022 0117 更新

ZZY看文章更新后绿色的字就好。

也许你都了解过，但可以在这里复习一下。

最近在用tensorflow跑cnn，之前一直用cpu版本，最近主机到了，开始学着装gpu版本。

上网看了一下，大多数tensorflow-gpu的安装分三部分：Anaconda的安装、CUDA+cudnn的安装、tensorflow-gpu的安装。

这些安装之间的版本要保持匹配，挺晕的。

但其实，利用Anaconda，是可以一步到位的。

本机配置：win10 gtx1060 

---

首先下载Anaconda： Anaconda | Individual Edition

安装之后记得添加环境变量（也可以不添加）：

---

FOR ZZY 

如何创建新的python环境：

在添加完环境变量后打开cmd（如果没有添加就打开Anaconda Prompt）

（1）查看conda环境： conda env list

  

（2）创建一个新环境：conda create -n env_name python=x.x

（有可能会出错Collecting package metadata (current_repodata.json): failed  试试关掉VPN）

（3）启动刚创建的环境：conda activate forzzy

(4)安装所需要的依赖库：conda install xxx

 安装完所有需要的库之后，就可以在PyCharm中使用这个库了：

打开PyCharm -> File -> Settings -> Project: XXX -> Python Interpreter ->  -> Add

这样就OK了。

---

 

 

下面不用看

 安装过后，打开Anaconda Prompt

先配置清华镜像！不然下载会很慢很慢很慢！

anaconda | 镜像站使用帮助 | 清华大学开源软件镜像站 | Tsinghua Open Source Mirror

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 --set show_channel_urls yes

输入

conda install tensorflow-gpu

可以发现会自动下载cuda+cudnn

输入y后开始（中途千万别关！）

出现这个说明安装OK 

如果提示有错误，那么..卸载了Anaconda然后重新安装吧...

---

用Pycharm加载Anaconda的环境库，并实验一下tensorflow-gpu能不能用：

在Pycharm中写入（如果你没装CPU版本的tensorflow，不写也可以）：

config = tf.ConfigProto()
config.gpu_options.allow_growth = True
with tf.Session(config=config) as session:
    # your training session

训练前/训练中GPU使用情况：

  

---

可以看出，这种方法是可行的。