正确安装 torch

#正确安装 torch| 来源: 网络整理| 查看: 265

step1：查看pytorch+cuda 版本 torch-scatter torch-sparse torch-cluster torch-spline-conv 这些关联包要与torch版本匹配。 import torch print(torch.__version__) print(torch.cuda.is_available()) torch.version.cuda 或者 pip list查看版本 step2：下载对应版本的包文件

比如我的是1.13.0，cu117版本，那么我就装1.13.0 cu117版本对应的那四个关联包，如下命令：

对应版本下载链接：https://data.pyg.org/whl/ pip install torch-scatter -f https://pytorch-geometric.com/whl/torch-1.13.0+cu117.html pip install torch-sparse -f https://pytorch-geometric.com/whl/torch-1.13.0+cu117.html pip install torch-cluster -f https://pytorch-geometric.com/whl/torch-1.13.0+cu117.html pip install torch-spline-conv -f https://pytorch-geometric.com/whl/torch-1.13.0+cu117.html pip install torch-geometric step3：注意torch-geometric的安装

若报错：AttributeError: module ‘torch_geometric.utils‘ has no attribute ‘scatter_‘ 直接搜这个错误，没有找到相关的内容，大多数解决方案是对于pytorch出现的报错，应该是用这个库的人不多。在Github上torch-geometric库的讨论区中也没有找到相关的内容。抱着试一试的心态，我直接搜这个函数*torch_geometric.utils.scatter_*的相关内容。在一篇介绍torch-geometric安装的博文中，我找到了一句关于torch-geometric版本的说明，举例介绍了“torch_geometric.utils.scatter_这个方法，在1.5及以后的版本就不支持了，所以代码中有用到这个方法的，要不就另写一个函数代替，要不就把版本退到1.4.3”。于是我尝试把torch-geometric退回到1.4.3版本：

pip install torch-geometric==1.4.3 step4：遇到的其他奇葩问题

OSError: Could not load shared object file: llvmlite.dll pip 卸载： numba 和 llvmlite 包然后： pip 安装

有人建议使用conda 安装，（慢且安装与其他的包冲突）本人未成功。改用pip 成功了！！！！ pip uninstall numba llvmlite pip install numba llvmlite step5：参考

———————————————— 参考链接：llvmlite.dll 错误解决

参考链接[1] torch-geometric 版本：https://blog.csdn.net/qq_40329272/article/details/111807624

参考链接[2] ：https://blog.csdn.net/qq_46031119/article/details/128153390

【本文地址】

公司简介

联系我们