RuntimeError: expected device cpu and dtype Byte but got device cpu and dtype Bool
RuntimeError: expected device cpu and dtype Byte but got device cpu and dtype Bool
如 CPU 上的 the issue I opened, I get the following error when running the Pytorch inverse-cooking 模型所述:
RuntimeError: expected device cpu and dtype Byte but got device cpu and dtype Bool
我已经尝试在笔记本电脑的英特尔 i7-4700HQ 8 线程和台式机 Ryzen 3700x 中 运行设置 demo.ipynb
文件。我在笔记本电脑上使用 Arch Linux,在台式机上使用 Manjaro。
当我在 Google Collabs GPU 上 运行 模型工作正常。
根据 demo.ipynb
文件,模型也应该能够 运行 在 CPU 上。有谁知道我是否必须调整任何参数才能使其正常工作?
如@iacolippo 和评论会话中所述,myDennisCode,问题确实出在依赖版本上。我有 torchvision==0.4.0
(这让我很困惑)和 torch==1.2.0
.
要解决此问题,只需安装 torch==0.4.1
和 torchvision==0.2.1
。
如 CPU 上的 the issue I opened, I get the following error when running the Pytorch inverse-cooking 模型所述:
RuntimeError: expected device cpu and dtype Byte but got device cpu and dtype Bool
我已经尝试在笔记本电脑的英特尔 i7-4700HQ 8 线程和台式机 Ryzen 3700x 中 运行设置 demo.ipynb
文件。我在笔记本电脑上使用 Arch Linux,在台式机上使用 Manjaro。
当我在 Google Collabs GPU 上 运行 模型工作正常。
根据 demo.ipynb
文件,模型也应该能够 运行 在 CPU 上。有谁知道我是否必须调整任何参数才能使其正常工作?
如@iacolippo 和评论会话中所述,myDennisCode,问题确实出在依赖版本上。我有 torchvision==0.4.0
(这让我很困惑)和 torch==1.2.0
.
要解决此问题,只需安装 torch==0.4.1
和 torchvision==0.2.1
。