位置: IT常识 - 正文

2021年Windows下安装GPU版本的Tensorflow和Pytorch(windows2022下载)

编辑:rootadmin
2021年Windows下安装GPU版本的Tensorflow和Pytorch

推荐整理分享2021年Windows下安装GPU版本的Tensorflow和Pytorch(windows2022下载),希望有所帮助,仅作参考,欢迎阅读内容。

文章相关热门搜索词:windows 1.0下载,windows 安装版,windows2020下载,windows2022下载,windows2022下载,2021版win10下载,windows2022下载,2021版win10下载,内容如对您有帮助,希望把文章链接给更多的朋友!

视频教程:2021年windows下安装GPU版本的Tensorflow和Pytorch_哔哩哔哩_bilibili

最近比特币的热潮慢慢褪去,显卡的价格也下来了,所以小伙伴们可以观察一下最近的行情,合适的时候可以入手几块显卡来搞深度学习。关注我的朋友大多数来自大作业怎么搞系列教程,大作业怎么搞系列使用tensorlfow训练了几个物体分类模型,在开源出的代码中我也基本给了大家我训练好的模型,有的朋友反应自己跑训练的过程的时候速度比较慢,这个原因大多都是因为大家是在CPU环境下来跑的,所以速度不是很快,如果使用GPU来跑训练的话,效率上差不多是CPU的10倍。

!!!注:深度学习的小伙伴请选择Nvidia(英伟达)的显卡

显卡的品牌基本可以分为两大阵营:AMD和Nvidia。如果是深度学习使用的话大家务必要选择Nvidia的显卡,AMD显卡性价比虽然比较高,但是其对深度学习的支持不是很好,所以大家购买之前一定要看清是不是英伟达的显卡,下面是英伟达的图标。

认识显卡(GPU)

购买显卡之前,我们以Nvidia的显卡为例,说明一下显卡的型号上面的数字都表示什么意思,影驰表示的是显卡的制造商,Geforce是显卡系列名称,GTX表示显卡的档次,数字一般是四位,前两位表示显卡的代数,16就表示是第16代显卡,中间的第三位数字表示的是显卡的性能级别,数字越大显卡的性能级别就相对越好,最后一位一般都是0,不用去管,有的显卡还有英文的后缀,SE表示阉割版、TI和Super表示增强的版本,比如1660TI就是在1660的基础上进行了增强,6G表示显卡的显存是6个GB,最后的比较炫酷的名字是制造商定的名字,这个一般就是个噱头,不需要太关注。

另外比较重要的是大家要学会看显卡的参数,分别是架构、工艺、流处理器、核心频率、显存频率、显存位宽、显存容量,比如这里是3090的显卡信息。

架构:相当于运行布局,布局越好跑的越流畅。工艺:制程越小精度越高,越能发挥更多性能。光栅以及流处理器:相当于劳动力,人越多执行力越强。核心频率:反应速度,相当于跑车百米提速效率。显存频率:相当于限速标志,决定了最大运行速度。显存位宽:相当于划线,决定了最大运行通道。显存容量:相当于道路限宽,决定了最大承载量。

拿绝地求生来说,这款游戏比较吃显存。因为游戏地图数据都加载在显存内,越是精致的3D画面数据量越多,需要的显存容量就越高。深度学习的话一般主要关注两个点,一个是显存的容量一个是cuda核心的数量,这两个指标越大相对就越好。

有台式机的小伙伴可以直接买单独的显卡插在主板上,注意要看自己的电源功率是否能够为显卡提供供电。

没有台式机的小伙伴也可以考虑买带有Nvidia显卡的笔记本,目标联想拯救者系列就是个不错的选择。

安装显卡驱动

首先第一步还是要安装显卡的驱动,显卡驱动的安装需要在官网下载驱动程序,首先大家需要在设备管理器中查看你的显卡型号,这里是我的显卡。

去官网下载对应你显卡的驱动程序:

下载之后执行程序并进行安装即可,为了防止不必要的情况出现,安装的时候请按照默认选项执行。

安装完毕之后重启电脑在cmd中输入nvidia-smi,输出下列信息则表示显卡驱动安装成功。

安装Anaconda

老朋友应该都知道Anaconda和Pycharm的使用方式,新朋友的话看下面这个教程就可以了

如何在pycharm中配置anaconda的虚拟环境_dejahu的博客-CSDN博客

安装完成之后在cmd中输入conda,如果输出下列信息则表示anaconda安装成功。

2021年Windows下安装GPU版本的Tensorflow和Pytorch(windows2022下载)

安装完毕之后请务必更新为国内的源,加快第三方库的下载速度,在cmd中执行下列指令即可。

conda config --remove-key channelsconda config --add channels https://mirrors.ustc.edu.cn/anaconda/pkgs/main/conda config --add channels https://mirrors.ustc.edu.cn/anaconda/pkgs/free/conda config --add channels https://mirrors.bfsu.edu.cn/anaconda/cloud/pytorch/conda config --set show_channel_urls yespip config set global.index-url https://mirrors.ustc.edu.cn/pypi/web/simple安装GPU版本的Tensorflow

GPU版本Tensorflow的安装

创建并激活虚拟环境

打开cmd,首先创建虚拟环境,分别输入下面两条命令,完成虚拟环境的创建和激活

conda create -n dejahu-tf python==3.7.3conda activate dejahu-tf安装

我们首先需要使用conda安装cuda和cudnn

conda install cudatoolkit=10.1conda install cudnn==7.6.5

然后使用pip指令安装gpu版本的tensorflow

pip install tensorflow-gpu==2.3.0测试GPU是否可用

现在在命令行中测试一下GPU是否可用,首先输入python进入python的解释器中

输入下面两条指令,如果输出为True则表示GPU可以使用

import tensorflow as tfprint(tf.test.is_gpu_available())

如果大家的输出和我这个类似那么你就可以愉快地使用GPU版本的tensorflow了

tensorflow如果安装的是GPU版本的则默认使用GPU,大家不需要在代码中指定,直接使用即可

安装GPU版本的Pytorch

GPU版本Pytorch的安装

创建并激活虚拟环境

打开cmd,首先创建虚拟环境,分别输入下面两条命令,完成虚拟环境的创建和激活

conda create -n dejahu-torch python==3.7.3conda activate dejahu-torch安装

我们这里使用conda进行gpu版本pytorch的安装,非常方便,直接在激活的虚拟环境中输入下列命令即可,这里安装的是最新版本的Pytorch

conda install pytorch torchvision torchaudio cudatoolkit=10.2

30系的朋友需要cuda11的支持,请执行下列命令

conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch -c conda-forge

如果需要指定版本号,请这样执行

conda install pytorch==1.5.0 torchvision==0.6.1 cudatoolkit=10.2测试GPU是否可用

现在在命令行中测试一下GPU是否可用,首先输入python进入python的解释器中

输入下面两条指令,如果输出为True则表示GPU可以使用

import torchprint(torch.cuda.is_available())

如果大家的输出和我这个类似那么你就可以愉快地使用GPU版本的Pytorch了

不过Pytorch会讲究使用device来指定GPU,需要大家通过to()方法做下转移

备用方案

另外受到网络的原因,可能需要大家手动安装cuda和cudnn,不过这种方法需要cuda和cudnn匹配,不推荐大家使用,命令如下

conda install cudatoolkit=10.1conda install cudnn==7.6.5

使用这些API,您可以开发分析工具,深入了解CUDA应用程序的CPU和GPU行为。CUPTI作为CUDA支持的所有平台上的动态库提供。请参阅CUPTI文档。

nvidia与cuda需要满足关系: https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html

cuda与cudnn需要满足关系: https://developer.nvidia.com/rdp/cudnn-archive

cuda历史版本下载连接: https://developer.nvidia.com/cuda-toolkit-archive

Tensorflow不同版本要求与CUDA及CUDNN版本对应关系: https://blog.csdn.net/u013066730/article/details/88845083

本文链接地址:https://www.jiuchutong.com/zhishi/299398.html 转载请保留说明!

上一篇:CSS给元素添加边框(样式、颜色、宽度)(css给div加边框)

下一篇:十大经典排序算法(下)(十大经典排序算法(动图演示C 实现))

  • 华为nova4e杂志锁屏怎么关闭(华为nova4e杂志锁屏的图片在哪里?)

    华为nova4e杂志锁屏怎么关闭(华为nova4e杂志锁屏的图片在哪里?)

  • 苹果xs充电速度(苹果xs充电速度测试)

    苹果xs充电速度(苹果xs充电速度测试)

  • 笔记本打不开机(戴尔笔记本打不开机)

    笔记本打不开机(戴尔笔记本打不开机)

  • 电脑8g运行内存够用吗(电脑8g运行内存可以开多少个应用)

    电脑8g运行内存够用吗(电脑8g运行内存可以开多少个应用)

  • 华为手机怎么消除杂音(华为手机怎么消除小艺)

    华为手机怎么消除杂音(华为手机怎么消除小艺)

  • qq免费挂件有哪些名字(qq免费挂件有哪些永久2022)

    qq免费挂件有哪些名字(qq免费挂件有哪些永久2022)

  • 注册微信必须要手机号码吗(注册微信必须要实名吗)

    注册微信必须要手机号码吗(注册微信必须要实名吗)

  • iphone经常跳出开机导航(苹果手机自动跳出)

    iphone经常跳出开机导航(苹果手机自动跳出)

  • 华为手机充电越充越少怎么解决(华为手机充电越来越慢怎么办)

    华为手机充电越充越少怎么解决(华为手机充电越来越慢怎么办)

  • 小米手机清除数据后果(小米手机清除数据后需要激活设备)

    小米手机清除数据后果(小米手机清除数据后需要激活设备)

  • 拼多多退货单号填错了还能修改吗(拼多多退货单号没有怎么办)

    拼多多退货单号填错了还能修改吗(拼多多退货单号没有怎么办)

  • 戴森v6充不进电(戴森v6充不进去电)

    戴森v6充不进电(戴森v6充不进去电)

  • 怎么把表格存到桌面(怎么把表格存到新建的文件夹里)

    怎么把表格存到桌面(怎么把表格存到新建的文件夹里)

  • 怎么卸载系统软件(如何卸载系统软件)

    怎么卸载系统软件(如何卸载系统软件)

  • iphone怎么贴膜(iphone如何贴膜)

    iphone怎么贴膜(iphone如何贴膜)

  • 苹果11系列是单卡还是双卡(苹果11系列是单层主板吗)

    苹果11系列是单卡还是双卡(苹果11系列是单层主板吗)

  • 华为mate30pro可以用5g吗(华为mate30pro可以用120w充电器吗)

    华为mate30pro可以用5g吗(华为mate30pro可以用120w充电器吗)

  • qq关注太频繁怎么解决(qq为什么关注操作过于频繁)

    qq关注太频繁怎么解决(qq为什么关注操作过于频繁)

  • 快手播放率降低什么原因(快手播放率突然特别低怎么回事)

    快手播放率降低什么原因(快手播放率突然特别低怎么回事)

  • 抖音怎么隐身看直播(抖音怎么隐身看主页)

    抖音怎么隐身看直播(抖音怎么隐身看主页)

  • 节日备忘录怎么制作(春节备忘录怎么做)

    节日备忘录怎么制作(春节备忘录怎么做)

  • websoc可以卸载吗(websoc是什么软件)

    websoc可以卸载吗(websoc是什么软件)

  • 优酷一个月vip怎么充值(优酷一个月vip在哪)

    优酷一个月vip怎么充值(优酷一个月vip在哪)

  • 手机虎牙怎么改头像(手机虎牙怎么改地址)

    手机虎牙怎么改头像(手机虎牙怎么改地址)

  • 微信通话图标怎么隐藏(微信通话图标怎么换)

    微信通话图标怎么隐藏(微信通话图标怎么换)

  • 超实用!JavaScript修改CSS变量,达到动态修改样式的目的(治疗孩子咳嗽小秘方,超实用)

    超实用!JavaScript修改CSS变量,达到动态修改样式的目的(治疗孩子咳嗽小秘方,超实用)

  • 加班费要计入个人账户吗
  • 税务会计常用会计科目
  • 现金流量表利息费用怎么算
  • 土地摊销全部计入成本吗
  • 物业费的税金是否向业主收取
  • 软件无形资产摊销年限是多久
  • 砖厂开票员的工作流程
  • 企业的培训费会计分录
  • 开票金额为什么会多出一分钱
  • 现金日记账是收付实现制吗
  • 电子承兑银行承兑
  • 购买空调的报销单怎么写
  • 凭证工本费的账务处理
  • 取得投资收益会导致经营现金流量小于
  • 工程预缴税款会计分录
  • 应征增值税不含税销售额是什么意思
  • 高亮!这些发票不能抵扣增值税
  • 资本公积盈余公积区别
  • 广告行业监管政策
  • 酒店收到物品损失怎么办
  • 百旺税控服务器管理系统
  • 总公司签订合同发票开具给分公司是否可以抵扣
  • 享受小型微利企业标准
  • 按揭购入固定资产怎么算
  • 个税申报月份有误如何调整?
  • 建筑行业会计成本会计工作流程
  • 日用品做账时放什么东西
  • win7怎么运行在哪里设置
  • 理财收益的增值怎么算
  • 如何在Windows 11上卸载更新
  • PHP 中dirname(_file_)讲解
  • 提前预支工资怎么报税
  • 营改增后的增值税税目税率表
  • 应交税金的会计科目
  • 应收票据贴现的性质是什么
  • 微前端架构实现
  • 计提坏账损失的范围
  • php输出数字
  • 备用金支出没有发票怎么办
  • 这是我见过最牛的人英语
  • html5开发的app有哪些
  • 专家坐诊信息
  • vue前端面试官常问的问题
  • 其他收益最后转入哪个科目
  • 固定资产折旧率如何计算
  • 应收账款收不回来做坏账处理分录
  • 对账小技巧
  • 一般纳税人的增值税税率
  • 零申报步骤
  • 权益法股权投资收益纳税调减
  • 委托代销受托方会计分录
  • 递延所得税如何理解
  • 视同销售的账务处理如何做?
  • 调整以前年度利息支出
  • 防疫物资采购会计处理
  • 以前年度损益调整在报表哪里体现
  • 本期缴纳上期应纳税额怎么填
  • win7系统如何打开
  • xp局域网
  • debian iso
  • Usrshuta.exe - Usrshuta是什么进程
  • 苹果电脑mac设备怎么删除
  • windows7回收站图标没了
  • win7怎么关闭自带键盘
  • 电脑操作系统win8
  • Win10预览版更新弹窗如何关闭
  • javascript+
  • linux中的shell编程
  • linux内核讲解
  • python编码大全
  • 用js设置颜色
  • android自定义view的三大流程
  • jquery查找指定元素
  • python else用法
  • 青岛地税局 局长
  • 留抵退税再提速
  • 深圳微信税务局登录不了
  • 税务稽查为什么不问证人
  • 长春市征收中心
  • 日本各界
  • 免责声明:网站部分图片文字素材来源于网络,如有侵权,请及时告知,我们会第一时间删除,谢谢! 邮箱:opceo@qq.com

    鄂ICP备2023003026号

    网站地图: 企业信息 工商信息 财税知识 网络常识 编程技术

    友情链接: 武汉网站建设