ubuntu OSError:libtorch_cuda.so:无法打开共享对象文件:没有这样的文件或目录

w6mmgewl  于 6个月前  发布在  其他
关注(0)|答案(1)|浏览(80)

enter image description here这个问题我已经困扰了一段时间,如果有人能帮我解决这个问题,我会非常感激。我使用的系统是Ubuntu与CUDA 12.0。
如上所述,我已经尝试卸载和重新安装torchaudio,但我仍然遇到同样的问题。

dzjeubhm

dzjeubhm1#

我也遇到过这种情况,如果你用conda安装了pytorch,它在识别哪个版本是正确的方面会有一些问题,我猜你有一个cpu版本,但是已经安装了pytorch-conda,并且想用它和gpu一起使用。
要解决这个问题,你要么运行(创建一个新的环境,否则它可能无法工作)

conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia

字符串
pytorch webpage上所述。
或者你可以告诉conda你想要的pytorch版本(here是所有可用的版本,你需要一个*cuda*而不是*cpu*
比如我的情况

conda install pytorch=2.1.0=py3.11_cuda11.8_cudnn8.7.0_0 -c pytorch


解决了这个问题。

相关问题