笔者正在自学深度学习相关知识,最近给电脑配上了pytorch的gpu版本,并且根据各种教程成功安装了cuda,正打算跑几个关系抽取模型的时候遇到了如标题所述的报错
在网上查询了一些资料后得出一些可能的错误:
1.可能是该电脑pytorch装的是CPU版本,针对这种情况,如果该模型可以用cpu版本训练,则需要在python执行命令后面加上 --gpu_ids -1。
E:\PythonCode\Net>python train.py --gpu_ids -1
类似于这样。
如果大家遇到相同的问题可以参考这篇文章进行修改
然而笔者安装的正是gpu版本,因此这种可能性被排除。
2.可能是该电脑上的pytorch和cuda版本不兼容导致的,这种