PyTorch ROCM安装后libc10_cuda.so缺失报错修复咨询

针对你遇到的libc10_cuda.so: cannot open shared object file警告，给你几个可行的解决步骤：

• 配置ROCM环境变量
  ROCM的库路径未被系统识别是常见原因。如果你的ROCM安装在默认路径/opt/rocm，执行以下命令临时生效：

  export LD_LIBRARY_PATH=/opt/rocm/lib:/opt/rocm/lib64:$LD_LIBRARY_PATH
  

  要是想永久生效，把上面的命令添加到你的~/.bashrc或~/.zshrc文件末尾，然后执行source ~/.bashrc（对应你的shell类型）刷新配置。

• 确保torch和torchvision版本匹配
  PyTorch 1.13.1+rocm5.2对应的torchvision版本应为0.14.1+rocm5.2，你可以手动指定版本重装torchvision：

  sudo pip3 install torchvision==0.14.1+rocm5.2 --extra-index-url https://download.pytorch.org/whl/rocm5.2
  

• 检查Python环境一致性
  报错路径显示torchvision安装在用户本地的Python目录（~/.local/lib/python3.10），但你用sudo pip3安装可能导致系统Python和用户本地Python环境不一致。建议去掉sudo，用用户级pip重新安装：

  pip3 install torch torchvision --extra-index-url https://download.pytorch.org/whl/rocm5.2 --user
  

• 手动定位并添加库路径
  如果上面的方法都没用，先找到libc10_cuda.so的位置：

  find ~/.local/lib/python3.10/site-packages/torch -name libc10_cuda.so
  

  假设找到的路径是/home/vic/.local/lib/python3.10/site-packages/torch/lib，就把这个路径加到LD_LIBRARY_PATH：

  export LD_LIBRARY_PATH=/home/vic/.local/lib/python3.10/site-packages/torch/lib:$LD_LIBRARY_PATH
  

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