RuntimeError
LHXhh opened this issue · comments
LHXhh commented
RuntimeError: Attempting to deserialize object on CUDA device 3 but torch.cuda.device_count() is 1. Please use torch.load with map_location to map your storages to an existing device.
end loading
MikePelton commented
The supplied script files (e.g. lego_test.sh) have a line:
gpu_ids='3'
...so we presume the authors have multiple GPU's and choose their index to be 3 - if you set that '3' to '0' and run again you should be fine.
LHXhh commented
Thank you very much!
…------------------ 原始邮件 ------------------
发件人: ***@***.***>;
发送时间: 2022年10月5日(星期三) 晚上6:39
收件人: ***@***.***>;
抄送: ***@***.***>; ***@***.***>;
主题: Re: [Xharlie/pointnerf] RuntimeError (Issue #57)
The supplied script files (e.g. lego_test.sh) have a line:
gpu_ids='3'
...so we presume the authors have multiple GPU's and choose their index to be 3 - if you set that '3' to '0' and run again you should be fine.
—
Reply to this email directly, view it on GitHub, or unsubscribe.
You are receiving this because you authored the thread.Message ID: ***@***.***>