RuntimeError: module must have its parameters and buffers on device cuda:0 (device_ids[0]) but found

RuntimeError: module must have its parameters and buffers on device cuda:0 (device_ids[0]) but found one of them on device: cuda:1

2020.7.20更新:
出现这种错误一般是由于显卡设置问题。如果是将多卡模型用单卡运行,一个简单的解决办法是:

		import os
		os.environ["CUDA_VISIBLE_DEVICES"]='2'

这个意思是显卡仅2号可使用。但是在:

self.model = torch.nn.DataParallel(self.model, device_ids=[0]) 

中,仅对程序可见的2号显卡在程序中的实际id是0,所以在程序中应该将显卡ID设为1。
同样的:
2, 3显卡可见后的编号实际应该是0, 1.

评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值