第一步:查看本机显卡配置是否支持安装gpu版本的tensorflow

安装gpu版的tensorflow需要安装cuda和cudnn,所以需要检查cuda是否支持本机gpu,需要计算能力大于3.1。

1.查看本机显卡配置,显卡配置为 NVIDIA Tesla P100-PCIE-12GB

如何用anaconda安装python 如何用anaconda安装tensorflow-gpu_深度学习


2.点击此链接https://developer.nvidia.com/cuda-gpus查看你的显卡型号是否支持安装tensorflow-gpu,可以看到我的显卡计算能力是6.0,支持安装

如何用anaconda安装python 如何用anaconda安装tensorflow-gpu_tensorflow_02


第二步:安装anaconda

1.下载地址:https://www.anaconda.com/products/individual

我下载的是Anaconda3-2020.02-Windows-x86_64,你们最好下载最新版本。

如何用anaconda安装python 如何用anaconda安装tensorflow-gpu_anaconda_03


如何用anaconda安装python 如何用anaconda安装tensorflow-gpu_cuda_04

一路next,注意下面一步不要勾选“Add Anaconda to my PATH enviroment variable”,等会我们手动配置环境变量。

如何用anaconda安装python 如何用anaconda安装tensorflow-gpu_tensorflow_05


2.配置anaconda环境变量

右击我的电脑——高级系统设置——环境变量——系统变量——双击path

如何用anaconda安装python 如何用anaconda安装tensorflow-gpu_anaconda_06


将以下三个路径加入,注意这里要换成你自己的安装路径。C:\Users\t-yaoguo\AppData\Local\Continuum\anaconda3

C:\Users\t-yaoguo\AppData\Local\Continuum\anaconda3\Scripts

C:\Users\t-yaoguo\AppData\Local\Continuum\anaconda3\Library\bin

下图是我的环境变量配置,最后记得点击确定,否则变量并没有添加上。

如何用anaconda安装python 如何用anaconda安装tensorflow-gpu_深度学习_07


3.测试anaconda是否安装成功

在cmd中输入conda -V,正常显示版本号说明已经配置好了。

如何用anaconda安装python 如何用anaconda安装tensorflow-gpu_如何用anaconda安装python_08


第三步:安装tensorflow-gpu

1.修改镜像源

Conda安装好以后,默认的镜像源国内很多网络环境下访问不稳定,下载速度慢,需要修改镜像源

找到目录 C:\Users<你的用户名> 下的配置文件.condarc,如果没有.condarc文件,则打开cmd命令窗口输入conda config --add channels https://pypi.douban.com/simple和conda config --set show_channel_urls yes,此时就能找到.condarc文件,修改内容如下:

如何用anaconda安装python 如何用anaconda安装tensorflow-gpu_cuda_09


2.创建conda环境

创建一个名为"tensorflow-gpu"的conda环境

如何用anaconda安装python 如何用anaconda安装tensorflow-gpu_深度学习_10


如何用anaconda安装python 如何用anaconda安装tensorflow-gpu_cuda_11


激活环境,这样就进入了刚刚创建的tensorflow-gpu环境。

如何用anaconda安装python 如何用anaconda安装tensorflow-gpu_anaconda_12


3.安装tensorflow-gpu

先通过以下命令升级pip指令:

如何用anaconda安装python 如何用anaconda安装tensorflow-gpu_cuda_13


安装GPU版本的tensorflow需要输入以下命令:

如何用anaconda安装python 如何用anaconda安装tensorflow-gpu_tensorflow_14


通过以下命令安装Anaconda基础包:

如何用anaconda安装python 如何用anaconda安装tensorflow-gpu_如何用anaconda安装python_15


第四步:安装CUDA Toolkit+cuDNN

1.查看需要安装的CUDA+cuDNN版本

具体安装的CUDA和cuDNN版本需要与安装的tensorflow-gpu相匹配,点击以下链接查看自己的tensorflow-gpu支持的CUDA和cuDNN版本:

https://www.tensorflow.org/install/install_windows#requirements_to_run_tensorflow_with_gpu_support 我安装的tensorflow-gpu是2.3.1版本的,支持的是CUDA Toolkit 10.1和cuDNN 7.6.3.

2.下载CUDA+cuDNN

下载CUDA:https://developer.nvidia.com/cuda-toolkit-archive

如何用anaconda安装python 如何用anaconda安装tensorflow-gpu_cuda_16下载cuDNN:https://developer.nvidia.com/rdp/cudnn-archive

如何用anaconda安装python 如何用anaconda安装tensorflow-gpu_如何用anaconda安装python_17


下载完成!

如何用anaconda安装python 如何用anaconda安装tensorflow-gpu_深度学习_18


3.安装CUDA Toolkit和cuDNN

如何用anaconda安装python 如何用anaconda安装tensorflow-gpu_如何用anaconda安装python_19


如何用anaconda安装python 如何用anaconda安装tensorflow-gpu_深度学习_20


勾选所有,

如何用anaconda安装python 如何用anaconda安装tensorflow-gpu_深度学习_21


一路通过即可。

接下来,解压“cudnn-10.1-windows10-x64-v7.6.3.30.zip”,将以下三个文件夹,拷贝到CUDA安装的根目录下。

如何用anaconda安装python 如何用anaconda安装tensorflow-gpu_cuda_22


如何用anaconda安装python 如何用anaconda安装tensorflow-gpu_cuda_23


4.配置环境变量

将下面四个路径加入到环境变量中,注意换成自己的安装路径。

如何用anaconda安装python 如何用anaconda安装tensorflow-gpu_深度学习_24


第五步:测试

1.查看是否使用GPU

如何用anaconda安装python 如何用anaconda安装tensorflow-gpu_如何用anaconda安装python_25


2.查看在使用哪个GPU

如何用anaconda安装python 如何用anaconda安装tensorflow-gpu_深度学习_26


以上,就是如何安装tensorflow-gpu,最最重要的就是弄清楚版本匹配的问题!!

第六步:出错

1.错误1:No module named ‘tensorflow’

多半是CUDA,cuDNN,tensorflow-gpu,python版本不匹配或者环境变量配置错误的问题。

2.错误2:ImportError:DLL load failed:找不到指定的模块。

这个可能是由于某种包没安装或没更新,具体问题具体分析,该安装的安装该更新的更新。