安装tensorflow-gpu版本一直是很多人心中的痛,鉴于此笔者将介绍一种方便的安装方式,祝你顺利脱坑。
前期准备 (1)系统需要:Win7或更高版本(64位,需VisioC++); (2)硬件支持:tensorflow-gpu版本需要硬件支持(NVIDIA显卡)。要求算力不低于3.0,驱动程序为410或更高。(笔者的显卡为1050Ti); (3)查看驱动版本:黑窗键入nvidia-smi,有如下显示: (如果不输出可能未添加环境变量,打开path添加C:\Program Files\NVIDIA Corporation\NVSMI)。 看到版本为419.35,符合安装要求,如果不符合去官网下载驱动。搭建环境 (1)Visio c++安装,去微软官网下载; (2)卸载所有的python相关的软件,安装Miniconda(Miniconda比Anoconda干净,避免包冲突造成的影响); (3)设置conda为国内镜像源(默认指向国外网站,下载速度堪忧); (4)打开Anoconda prompt键入conda install tensorflow-gpu==2.x.x(一键安装省时省力,具体型号按需填写)等待安装完毕即可。验证安装 (1)打开Anoconda prompt,下载jupyter notebook 两种方法: conda install jupyter notebook pip install jupyter notebook (2)下载完成后键入jupyter notebook 新建文件输入: import tensorflow as tf tf.__version__(输出版本号) tf.test.is_gpu_available()(查看是否可使用GPU)执行上述程序后如果输出:
‘2.x.x’ Ture恭喜你安装成功!!!