将pytorch的网络等转移到cuda
作者：aleien1 发布时间：2023-08-10 08:33:46
标签：pytorch,网络,cuda
神经网络一般用GPU来跑，我们的神经网络框架一般也都安装的GPU版本，本文就简单记录一下GPU使用的编写。
GPU的设置不在model，而是在Train的初始化上。
第一步是查看是否可以使用GPU
self.GPU_IN_USE = torch.cuda.is_available()
就是返回这个可不可以用GPU的函数，当你的pytorch是cpu版本的时候，他就会返回False。
然后是：
self.device = torch.device('cuda' if self.GPU_IN_USE else 'cpu')
torch.device是代表将torch.tensor分配到哪个设备的函数
接着是，我看到了一篇文章，原来就是将网络啊、数据啊、随机种子啊、损失函数啊、等等等等直接转移到CUDA上就好了！
于是下面就好理解多了：
转移模型：
self.model = Net(num_channels=1, upscale_factor=self.upscale_factor, base_channel=64, num_residuals=4).to(self.device)
设置cuda的随机种子：
torch.cuda.manual_seed(self.seed)
转移损失函数：
self.criterion.cuda()
转移数据：
data, target = data.to(self.device), target.to(self.device)
pytorch 网络定义参数的后面无法加.cuda()
pytorch定义网络__init__()的时候，参数不能加“cuda()", 不然参数不包含在state_dict()中，比如下面这种写法是错误的
self.W1 = nn.Parameter(torch.FloatTensor(3,3), requires_grad=True).cuda()
应该去掉".cuda()"
self.W1 = nn.Parameter(torch.FloatTensor(3,3), requires_grad=True)
来源：https://blog.csdn.net/weixin_42128941/article/details/103048866
0
投稿猜你喜欢
Check In/Out功能简介：该功能是专门针对多用户管理而预设的。即多用户使用多帐号管理同一站点。- 【先锋缓存类】Ver2004作者:孙立宇、apollosun、ezhonghua官方网站:http://www.lkstar.com 技术支
- 一个项目开发完毕后总有一种想法，就是生成可执行文件，总不能一直用python xxx执行吧。以下操作同时适用于windows和Linux下的
- 表中主键必须为标识列，[ID] int IDENTITY (1,1)1.分页方案一：(利用Not In和SELECT TOP分页)语句形式：
- c shell perl php下的日期时间转换: 秒数与人类可读日期 scalar localtime 与 seconds since `
- 【eval()函数】JavaScript有许多小窍门来使编程更加容易。其中之一就是eval()函数，这个函数可以把一个字符串当作一个Java
- 1、安装 python3sudo apt install python32、卸载 python2.7 （可选）sudo apt remove
- ASP从发布至今已经7年了，使用ASP技术已经相当成熟，自从微软推出了ASP.NET之后就逐渐停止了对ASP版本的更新。但是由于有很多人仍然
- A 定义数组有两种方式：DIM和REDIM。DIM定义的是固定个数、数据类型的数组；而REDIM则不同，它可以定义不同类型的数据，也可以定义
- 可在图片连接后面加上宽和高，如：width="150" height="50"<h
- overflow:hidden这个CSS样式是大家常用到的CSS样式，但是大多数人对这个样式的理解仅仅局限于隐藏溢出，而对于清除浮动这个含义
- access中可以将文本中的数据轻松导入表中，mysql中用起来没那么方便，其实起来也很简单。首先将数据记录按行处理好用特定的字符分开如：“
- 写入：1:把gif图像文件读入内存（一个变量strTemp）。2:写入数据库。Dim binTmp() As
- 多个字段模糊查询， 括号中的下划线是双下划线，双下划线前是字段名，双下划线后可以是icontains或contains,区别是是否大小写敏感
- 大家都知道一些论坛的标题有高亮显示功能，在这里我不讨论也不研究论坛普遍的实现方法，下面是我的实现方法：实现思路：把要高亮显示的标题加上特定标
- 相信用过Gmail的人都知道Gmail有一个草稿自动保存的功能，每过一段时间，Gmail都会自动保存邮件草稿，这样在一些突 * 况下就能快速地
- 本文实例讲述了Python高级编程之继承问题。分享给大家供大家参考，具体如下：多继承问题1.单独调用父类： 一个子类同时继承自多个父类，又称
- <％@LANGUAGE="xxx" CODEPAGE="936"％>一般又分为<％
- RabbitMQ 6种工作模式对RabbitMQ 6种工作模式(简单模式、工作模式、订阅模式、路由模式、主题模式、RPC模式)进行场景和参数
- 前提条件：1.安装好Wampserver64（版本不限）2.Wampserver64软件启动后 变为绿色如：3.在数据库里面创建好名为&am
