ludwig 运行时错误：预期所有Tensor都在同一个设备上，但至少发现了两个设备，cuda:0 和 cpu!(在检查方法 Package 器中的参数 mat1 的参数时...)

2jcobegt 于 2个月前发布在其他

关注(0)|答案(2)|浏览(27)

描述bug

我按照"如何贡献"的步骤，在本地设置了Ludwig进行开发。当我运行pytest时，我得到了以下错误信息：

环境信息(请填写以下信息):

2条答案

嘿，@karanjakhar,一些测试在GPU上运行时存在已知问题。我们正在进行PR来修复这些测试：
#3578
作为临时解决方法，你可以尝试在CPU上运行这些测试吗？它应该可以像这样工作：

CUDA_VISIBLE_DEVICES="" pytest ...

cc @jeffkinnison

大多数问题都可以通过你的建议解决。但是我现在遇到了这个问题：