安装ROCm版本PyTorch后Pip无法识别及导入报错如何解决?
ROCm平台PyTorch安装异常修复方案
报错1:导入torch提示缺少libtinfo.so.5
报错信息:ImportError: libtinfo.so.5: cannot open shared object file: No such file or directory
修复方法:
该错误由系统缺失ncurses5兼容库导致,根据你使用的Linux发行版执行对应安装命令即可:
- Debian/Ubuntu系列:
sudo apt install libncurses5 - RHEL/CentOS/Fedora系列:
sudo dnf install ncurses-compat-libs - Arch系列:
sudo pacman -S ncurses5-compat-libs
安装完成后重新导入torch即可正常加载。
报错2:安装torchvision后提示找不到torch模块
报错信息:ModuleNotFoundError: No module named 'torch'
修复方法:
该错误是单独安装torchvision时,pip默认拉取的torchvision依赖版本为CUDA适配版,会覆盖已安装的ROCm版本torch导致异常,按以下步骤操作:
- 先卸载当前环境中所有不匹配的PyTorch相关包:
pip uninstall torch torchvision torchaudio -y - 一次性执行ROCm版本对应的完整安装命令,同时安装torch、torchvision、torchaudio三个组件,不要分开单独安装各组件,避免版本不匹配。
- 安装完成后确认当前使用的Python环境和安装包的环境一致,避免多Python环境导致的包路径不匹配问题。
内容的提问来源于stack exchange,提问作者Matheus Borges
相关产品推荐
相关产品推荐

