安装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
恭喜你安装成功!!!