前提:一定要有独立显卡,而且计算能力在3.0以上(最好3.5以上)
安装TensorFlow-gpu版本需要安装CUDA和cuDNN,但是之前你要做好一系列前期准备。
确认步骤:
- 查看显卡及计算能力
- 查看显卡对应的CUDA版本
- 查看CUDA对应的cuDNN对应的版本
- 根据CUDA和cuDNN版本确认可以安装的TensorFlow-gpu版本
我们一步一步来:
1.查看显卡属性及计算能力
1、在电脑桌面,右击“此电脑”,点击“属性”
2、在打开界面,选择“设备管理器”
3、先在“设备管理器”界面,点击“显示适配器”,显卡的型号就会显示出来。
一般 Intel开头的为集成显卡,
一般 AMD、NVIDIA开头的为独立显卡。
查看计算能力:官方链接
满足3.0以上的计算能力,OK,你可以使用TensorFlow-gpu版本了。不行的话,加显卡或者换电脑!
2.查看显卡驱动对应的CUDA版本并且下载安装
这个CUDA版本也不能随便选择的,这得看你的显卡驱动能撑起哪个CUDA版本的运算。先找显卡驱动版本:控制面板->搜索“NVIDIA”->NVIDIA控制面板