search

xiangking / ark-nlp

A private nlp coding package, which quickly implements the SOTA solutions.
Apache License 2.0
310 stars 65 forks source link

没有cuda #61

Open YangFW opened 2 years ago

YangFW commented 2 years ago

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

YangFW commented 2 years ago

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

xiangking commented 2 years ago

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

YangFW commented 2 years ago

我这样设置过，也是报同样的错误，定位到的报错位置是这样的 image 为什么都设置了cpu，还会去调用torch.cuda.Stream()，难道需要我安装pytorch 的gpu版本吗？

xiangking commented 2 years ago

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

YangFW commented 2 years ago

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

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

xiangking commented 2 years ago

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