前言
这个错误网上大多数说是因为torch torchvision cuda以及python版本都是有兼容关系的,经过查找材料,原因猜测是因为torch版本过高,但是torchvision版本过低或者不兼容导致的;常用的解决方法是降低torch版本等,这里使用另外一种方法。
解决方法
找到CUDA文件路径,修改如下,
export PATH=/usr/local/cuda-11.8/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-11.8/lib64 #这句命令需要根据具体情况进行修改,比如你的cuda版本是11.7,则将命令修改为export LD_LIBRARY_PATH=/usr/local/cuda-11.7/lib64