Ubuntu+CUDA12.0环境下libtorch_cuda.so文件缺失问题求助

具体解决步骤 #

• 确认PyTorch与CUDA版本匹配
  先运行以下命令查看PyTorch绑定的CUDA版本：

  python -c "import torch; print(torch.version.cuda)"
  

  版本必须和你系统的12.0一致。如果不匹配，卸载现有PyTorch和torchaudio，重装对应CUDA12.0的版本：

  pip uninstall torch torchaudio -y
  pip install torch torchaudio --index-url https://download.pytorch.org/whl/cu121
  

  注：PyTorch官方未单独提供cu120包，cu121的包兼容CUDA12.0版本。

• 配置CUDA环境变量
  将CUDA的库路径添加到系统动态链接库路径：

  1. 编辑~/.bashrc文件，加入以下内容：

     export LD_LIBRARY_PATH=/usr/local/cuda-12.0/lib64:$LD_LIBRARY_PATH
     export PATH=/usr/local/cuda-12.0/bin:$PATH
     

  2. 执行source ~/.bashrc使配置生效，再用echo $LD_LIBRARY_PATH确认路径已添加。

• 检查libtorch_cuda.so文件是否存在
  用命令查找系统中的该文件：

  sudo find / -name "libtorch_cuda.so" 2>/dev/null
  

  如果找到，将其所在目录添加到LD_LIBRARY_PATH；若未找到，说明PyTorch安装不完整，重新执行前面的安装命令。

• 虚拟环境检查（若使用）
  如果你用了虚拟环境，必须激活环境后再执行安装操作，同时确保环境变量在虚拟环境中能正常加载。

内容的提问来源于stack exchange，提问作者Ivan Wang