Windows下快速安装配置TensorFlow(GPU)

硬件配置

具备NVIDIA的显卡

其他配置不要太坑就行

系统环境

win7/win10(都测试过了)

安装步骤

1.下载Anaconda，安装···(一路next即可)，安装结束时会提示是否安装VS CODE，建议安装，挺好用。

2.安装完anaconda后，在开始中打开anaconda的Anaconda Prompt。

此时进入了anaconda的(base)环境，在此新建一个tensorflow(gpu)的虚拟环境，输入命令：

conda create -n tensorflow-gpu python=3.6

此时，已经建立好(tensorflow-gpu)的虚拟环境，使用命令：

conda activate tensorflow-gpu

切换到(tensorflow-gpu)环境，可以看到环境已经发生变化(base)->(tensorflow-gpu)

在此新环境下安装tensorflow-gpu，输入命令

conda install tensorflow-gpu

到此，已经开始下载安装tensorflow和其依赖的一些包。但是cuda和cudnn可能下载失败(网不好的话)。这时候就需要手动下载安装对应tensorflow的cuda和cudnn版本。

cuda下载地址：https://developer.nvidia.com/cuda-toolkit-archive

cudnn下载地址：https://developer.nvidia.com/rdp/cudnn-archive // 需要注册登录

到此，已经安装好了。

3.验证是否安装成功

在(tensorflow-gpu)虚拟环境下输入python进入python环境，import tensorflow看是否报错，如果没有报错则安装成功。

OK，可以开始愉快的使用tensorflow了！

友情链接

https://www.jianshu.com/p/582ee249012e // 在VSCODE下使用tensorflow

https://blog.csdn.net/geyunfei_/article/details/78782804 // tensorflow入门教程