Tensorflow-GPU安装教程

安装tensorflow-gpu版本一直是很多人心中的痛,鉴于此笔者将介绍一种方便的安装方式,祝你顺利脱坑。

  1. 前期准备
    (1)系统需要:Win7或更高版本(64位,需VisioC++);
    (2)硬件支持:tensorflow-gpu版本需要硬件支持(NVIDIA显卡)。要求算力不低于3.0,驱动程序为410或更高。(笔者的显卡为1050Ti);
    (3)查看驱动版本:黑窗键入nvidia-smi,有如下显示:
    (如果不输出可能未添加环境变量,打开path添加C:\Program Files\NVIDIA Corporation\NVSMI)。
    在这里插入图片描述
    看到版本为419.35,符合安装要求,如果不符合去官网下载驱动。
  2. 搭建环境
    (1)Visio c++安装,去微软官网下载;
    (2)卸载所有的python相关的软件,安装Miniconda(Miniconda比Anoconda干净,避免包冲突造成的影响);
    (3)设置conda为国内镜像源(默认指向国外网站,下载速度堪忧);
    (4)打开Anoconda prompt键入conda install tensorflow-gpu==2.x.x(一键安装省时省力,具体型号按需填写)等待安装完毕即可。
  3. 验证安装
    (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

恭喜你安装成功!!!

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值