没有cuda

[YangFW]

没有安装cuda，如何调用cpu训练，现在报如下错误：
AssertionError: Torch not compiled with CUDA enabled

[YangFW]

模型设置时不带device参数，或者直接设置为None，训练时都报上述同样的错
model = Task(dl_module, optimizer, CasRelLoss(),device=None)

[xiangking]

您好，需要设置device="cpu"，设置为None时会去检测环境自动选择

[YangFW]

我这样设置过，也是报同样的错误，定位到的报错位置是这样的

为什么都设置了cpu，还会去调用torch.cuda.Stream()，难道需要我安装pytorch 的gpu版本吗？

[xiangking]

casrel的代码吗，这部分确实存在这个问题，为了速度，所以设置了torch.cuda.Stream

[YangFW]

casrel的代码吗，这部分确实存在这个问题，为了速度，所以设置了torch.cuda.Stream

对，是的，那目前我没有显卡的情况下，还能用这块的函数吗，还是必须要换掉？
PRGC 的有这个问题吗

[xiangking]

PRGC没用到这部分代码应该是可以，casrel后续我们会对该问题进行优化