Pytorch
最好的深度学习框架,没有之一
IMU_YY
这个作者很懒,什么都没留下…
展开
-
RuntimeError: CUDA error: no kernel image is available for execution on the device
记录一下这个bug0x01 bug的由来 疫情原因,大家都没有返校,而我回来了,正赶上服务器需要重装,这个重任落到了我的肩膀上——如何走上运维之路0x02 环境ubuntu 16.04(应该是)显卡型号Tesla k40mpython版本:3.6.10pytorch版本:1.5.00x03 bug描述 tensor可以放到gpu,但是不可以使用。从下图可以看到,我把a放到cuda上是没有问题的,但是在查看a的内容,也就是使用a的时候出现了RuntimeError: CUDA .原创 2020-07-20 12:26:49 · 30452 阅读 · 16 评论 -
torch.sigmoid、torch.nn.Sigmoid和torch.nn.functional.sigmoid的区别
review代码的时候发现我使用的是torch.sigmoid,pycharm标黄cannot find reference 'sigmoid' in '__init__.py'于是产生了这样的疑问,这三者的区别是啥。0x00. 官网解释torch.sigmoid 我们可以看到,这是一个方法,拥有Parametrs和Returns。torch.nn.Sigmoid 可以...原创 2019-05-21 17:50:35 · 20940 阅读 · 2 评论 -
RuntimeError: CUDA error: initialization error
问题描述:使用PyTorch的DataLoader,讲num_workers改为非零数字后, 运行程序报错RuntimeError: CUDA error: initialization error解决方法:不可在DataLoader或DataSet内将任何数据放到CUDA上,而是等到程序运行出DataLoader之后(也就是到了train里的时候)将数据放到CUDA上。原因猜测: D...原创 2019-05-31 16:23:30 · 15299 阅读 · 12 评论