当前位置: 代码迷 >> 综合 >> RuntimeError: Expected a ‘cuda‘ device type for generator but found ‘cpu‘
  详细解决方案

RuntimeError: Expected a ‘cuda‘ device type for generator but found ‘cpu‘

热度:95   发布时间:2023-11-19 23:05:54.0
RuntimeError: Expected a 'cuda' device type for generator but found 'cpu'

使用的1.9版本的pytorch太高了,换成pytorch1.8之后就行了。

  相关解决方案