Pip未识别带有ROCm安装的PyTorch

Question

在PyTorch website上，它列出了用于ROCm版本安装的两个命令块。第一个是安装torch本身，运行良好，但当我尝试导入它时，会显示此消息

ImportError: libtinfo.so.5: cannot open shared object file: No such file or directory

另外，当尝试使用第二个命令块安装torchvision软件包时，它会显示类似的错误

ModuleNotFoundError: No module named 'torch'

这仅适用于ROCm计算平台。使用CUDA安装工作正常，但不幸的是我没有NVidia GPU

Answer 1

我相信这是一个尚未修复的错误。您可以在与libtaichi_core.so相同的文件夹中创建名为libtinfo.6.so到/usr/lib/libtinfo5.so的本地符号链接

这就应该解决了,