pin_memory报错解决:RuntimeError: cannot pin ‘CUDAComplexFloatType‘ only dense CPU tensors can be pinned

这个是由于你的数据类默认使用cuda类的tensor,但是读数据的时候应该使用cpu类型的tensor,在forward的时候再将数据变为cuda类型tensor。解决方法就是将数据类中cuda类型的tensor改为cpu类型的tensor,在模型forward的时候再变为cuda类型的tensor


本文来自互联网用户投稿,文章观点仅代表作者本人,不代表本站立场,不承担相关法律责任。如若转载,请注明出处。 如若内容造成侵权/违法违规/事实不符,请点击【内容举报】进行投诉反馈!

相关文章

立即
投稿

微信公众账号

微信扫一扫加关注

返回
顶部