win10 anaconda 安装Tensor Flow2.x(GPU版)
根据自己的安装过程做一下记录,文末附有安装过程中学习过的文章链接
Tensor Flow版本的选择
首先决定是GPU版本还是CPU版本
判断条件 | 结果 |
---|---|
显卡是不是N卡(NVIDIA) | 是=判断算力;否=CPU版本 |
显卡是N卡(NVIDA) | 大于等于3.5=GPU版本;小于3.5=CPU版本 |
如何判断显卡是否是N卡(NVIDIA):
此电脑-右键属性-设备管理器-显示适配器
图中显示适配器下面即可查看显卡型号(如本人的独显为NVIDIA GTX 1050Ti)
如何判断显卡算力:
具体信息可前往NVIDIA官网查看
查看显卡算力-NVIDIA官网链接
安装显卡驱动、CUDA、cuDNN
在NVIDIA官网下载新版的显卡驱动
显卡驱动下载链接
查看Tensor Flow与CUDA、cuDNN的版本对应关系
查看Tensor Flow与CUDA、cuDNN版本对应关系链接
下载所需要的CUDA
CUDA下载链接
安装CUDA时选择自定义安装,不勾选Visual Studio Integration,否则可能会报错(可能)
安装路径
下载所需要的cuDNN(这里需要登录NVIDIA账号,可能会慢一些)
cuDNN下载链接
cuDNN解压后会有三个文件夹