完美解决pytorch多线程问题：Cannot re-initialize CUDA in forked subprocess. To use CUDA with multiprocessing

在pytorch，dataloader里设置了num_workers > 1的时候, 会出现这个错误。

一般网上RuntimeError: Cannot re-initialize CUDA in forked subprocess解决方法之一_NLOS的博客-CSDN博客解决方法，加上一行设定。

然而，我这里如果加上这行代码，会导致代码卡死在dataloader里面。

后面发现，代码里面设定了这样的几行。

if cuda:torch.set_default_tensor_type('torch.cuda.FloatTensor')
else:torch.set_default_tensor_type('torch.FloatTensor')

导致了，在dataloader的时候，tensor就会被设定为cuda上的变量，多线程的时候就会出问题，当注释掉这几行之后，就不会报错了。

完美解决pytorch多线程问题：Cannot re-initialize CUDA in forked subprocess. To use CUDA with multiprocessing相关推荐

完美解决Pytorch在Pycharm没有代码提示的问题
在Pytorch的旧版本中,没有__init__.pyi这个文件(Pycharm就从__init__.pyi读取函数的声明以及参数类型),因此在Pycharm中torch.sum.torch.abs等 ...
android 置灰不可点击,Android Studio 运行按钮灰色的完美解决方法
Android Studio 运行按钮灰色的完美解决方法今天新建项目的时候突然发现编译后运行按钮为灰色. 解决方案:第一步:点击图中的Add Configuration,出来如下界面第二步:点+号 ...
python database is locked_sqlite遇到database is locked问题的完美解决
这两天在项目中用大强度大频率的方法测试时遇到sqlite报database is locked的问题, 分析下来原因是sqlite对数据库做修改操作时会做(文件)锁使得其它进程同一时间使用时会报该错误 ...
sqlite遇到database is locked问题的完美解决
这两天在项目中用大强度大频率的方法测试时遇到sqlite报database is locked的问题, 分析下来原因是sqlite对数据库做修改操作时会做(文件)锁使得其它进程同一时间使用时会报该错误 ...
完美解决绝大部分域外名的学术网站访问缓慢问题
小白必会,完美解决绝大部分域外名的学术网站访问缓慢问题文章目录一.国内访问国外域外名的学术网站为什么很慢? 二.解决 DNS 污染的两种方法: 1.方法一(自动修改,==最推荐==) 2.方法二( ...
CornerNet代码学习之pytorch多线程
Cornernet代码之pytorch多线程学习源码剖析 main() train() 页锁定内存守护线程 init_parallel_jobs().pin_memory() 信号量附录-源码内 ...
js手机键盘遮挡_完美解决手机网页中输入框被输入法遮挡的问题
之前要做一个弹出对话框,填写信息,发现在手机上看的时候,较后的输入框在填写信息时,输入框被输入法遮挡,只能盲填. 前提 1.弹出的对话框用display:fixed定位的 2.对话框大小固定解决办法 ...
UITableViewCell 左侧滑动删除按钮添加图片（不完美解决）
*需求:给cell左侧滑动删除按钮添加图片 //目前的解决方法链接: https://pan.baidu.com/s/1kVE5gMF 密码: zaph *装态:还在解决网上查过资料一直没好的解决 ...
html百度转到顶部 fixed,科技常识：完美解决安卓端百度浏览器屏蔽fixed悬浮元素的问题...
今天小编跟大家讲解下有关完美解决安卓端百度浏览器屏蔽fixed悬浮元素的问题 ,相信小伙伴们对这个话题应该有所关注吧,小编也收集到了有关完美解决安卓端百度浏览器屏蔽fixed悬浮元素的问题的相关资料 ...
docker 挂载目录_完美解决：Docker部署SpringBoot项目后图片无法访问和上传，3招搞定！...
以前使用FTP在Linux中使用java -jar xxx.jar部署SpringBoot项目时,由于在项目中指定了主机文件存放路径映射,可以直接访问服务器的文件. 但是最近入坑Docker,将Spr ...

完美解决pytorch多线程问题：Cannot re-initialize CUDA in forked subprocess. To use CUDA with multiprocessing

完美解决pytorch多线程问题：Cannot re-initialize CUDA in forked subprocess. To use CUDA with multiprocessing相关推荐

最新文章

热门文章