文章目录
window10 下 anaconda 安装 tensorflow-gpu
安装tensorflow的软硬件要求
官方给出了 tensorflow-gpu 的软硬件要求,可以参考链接自行查询
链接:tensorflow官网提出的软硬件要求
这里提供一张官网的截图,截图时间是 2019年5月7日 。由于tensorflow也在不断更新,具体的软硬件要求请以官网给出的为准。
根据硬件要求,计算机的显卡需要满足一定的计算能力(大于等于3.5),比如我的显卡是 GTX970,根据列表,GTX970 的计算能力是 5.2,满足要求。如果计算能力不行,估计就不能继续了。
除了硬件有要求,软件也有要求,根据截图上的说明,如果想要成功运行 tensorflow,需要:
- 显卡驱动程序需要410.x或者更高的版本
- 安装 CUDA 10.0 工具包
- cuDNN 7.4.1 版本或者更高
- TensorRT 5.0(可选,实际上我没装,因为我暂时用不上)
- 其他。网上有人说电脑得先安装VS,我的电脑是安装了 Visual Studio 2017的。
更新显卡驱动
电脑当前的显卡驱动版本可以从 Nvidia 控制面板查看(桌面右击)
我的版本是398.75,要求是需要410.x或者更高,我这个太低了。去官网下了个显卡对应的最新的驱动程序,最新的是430.39,然后安装。我是直接双击430.39安装的,并没有卸载之前的显卡驱动。