如何进行PyTorch的GPU使用

这期内容当中小编将会给大家带来有关如何进行PyTorch的GPU使用,文章内容丰富且以专业的角度为大家分析和叙述,阅读完这篇文章希望大家可以有所收获。

创新互联为客户提供专业的成都网站建设、网站建设、程序、域名、空间一条龙服务,提供基于WEB的系统开发. 服务项目涵盖了网页设计、网站程序开发、WEB系统开发、微信二次开发、手机网站制作设计等网站方面业务。

GPU如何使用

一般我们只有一个GPU,所以使用GPU训练需要输入:

        device = torch.device("cuda:0")      # 声明用到的是第0块GPU

        model = Model(input_size, output_size)      # Model是网络名,一般是类名。实例化Model

        model = model.to(device)      # 将实例化的model送入GPU中。

上面这一步将model送入了GPU,我们还需要把输入数据、真值也送入GPU中,于是也必须有:

        inputs, labels = inputs.to(device), labels.to(device)

注意:

    在pytorch里,inputs.to(device)和inputs.cuda()效果是一样的。

    在pytorch里, 对于tensor,b=a.to(device)和a.to(device)是不一样的,前者会让b进入GPU,而后者并不会让a进入GPU。

                            对于model,b=a.to(device)和a.to(device)是一样的,b和a都会进入GPU。

上述就是小编为大家分享的如何进行PyTorch的GPU使用了,如果刚好有类似的疑惑,不妨参照上述分析进行理解。如果想知道更多相关知识,欢迎关注创新互联行业资讯频道。


文章题目:如何进行PyTorch的GPU使用
浏览路径:http://bzwzjz.com/article/jjpood.html

其他资讯

Copyright © 2007-2020 广东宝晨空调科技有限公司 All Rights Reserved 粤ICP备2022107769号
友情链接: 响应式网站设计方案 成都网站设计 成都网站建设 企业网站建设 四川成都网站制作 宜宾网站设计 定制网站制作 成都企业网站制作 成都网站建设推广 成都网站设计 成都定制网站建设 自适应网站设计 营销型网站建设 营销型网站建设 成都网站制作 成都网站设计 成都网站制作 成都网站建设 网站建设公司 成都网站建设 成都网站设计 成都商城网站制作