运行百度飞桨的代码,报以下错误,并给出提示,是由于cuda没有配置好导致的报错。
但是电脑是安装过pytorch,并运行成功过gpu代码的,所以应该配置好了cuda和cudnn。
有人说需要再安装一遍,我觉得不对,应该是pytorch的cuda安装在某个目录下,但是这个目录没有被飞桨检查到,而非不存在cuda。
解决步骤(linux环境下:):
1. 找到pytorch的cuda目录,输入指令:
find / -name 'cudnn'
2. 配置好路径,让飞桨找到
export LD_LIBRARY_PATH='/usr/local/anaconda3/envs/rasa-gpu/lib/'
问题解决!如果对您有帮助,请点赞谢谢!!
欢迎分享,转载请注明来源:内存溢出
评论列表(0条)