torch安装教程

首先确定机器使用的cuda版本nvcc -V：

例如本服务器使用的时cuda11.1版本

创建一个conda环境：

conda create -n new_env python=3.8conda activate new_env

安装pytroch环境：
conda install pyg -c pyg
该命令会把一些常用的torch以及关联的工具包都安装到机器，验证是否正确安装：

如果发生错误：

需要自行去官网查找正确的版本：Previous PyTorch Versions | PyTorch

安装torch-geometric：
torch-geometric要和torch版本配套，登录网站链接: 链接查看对应的版本：


下载正确版本的组件，例执行pip install torch-sparse==0.6.12，其他的组件也一样

最后安装torch-geometric： pip install torch-geometric
失败了可以尝试加上版本号pip install torch-geometric==2.0.4

验证torch-geometric：

成功导入即可

来源地址：https://blog.csdn.net/Newbg6/article/details/129422790