【踩坑】Linux下配置torch-geometric

最新推荐文章于 2025-08-25 20:28:31 发布

原创

最新推荐文章于 2025-08-25 20:28:31 发布 · 2.3k 阅读

4 ·

CC 4.0 BY-SA版权

文章标签：

#torch #conda

搞了整整一天终于配置好了。。。= =

1、创建虚拟环境

conda create -n xxx python=3.6.11
conda activate xxx

2、conda install pytorch=1.4 pytorch官网的指令安装的是最新版本，不适配！！！后面会安装不上torch-geometric库

3、两种方法：（我是使用方法二成功的）

使用官方文档里的方法无脑依次进行检查并安装torch-geometric等库。。。
官方方法无效，使用官方文档方法直到验证完版本信息后，不再使用文档内的安装语句，使用如下语句改成自己需要的版本后，进行安装。

pip install torch-scatter==latest+${CUDA} -f https://pytorch-geometric.com/whl/torch-1.5.0.html
pip install torch-sparse==latest+${CUDA} -f https://pytorch-geometric.com/whl/torch-1.5.0.html
pip install torch-cluster==latest+${CUDA} -f https://pytorch-geometric.com/whl/torch-1.5.0.html
pip install torch-spline-conv==latest+${CUDA} -f https://pytorch-geometric.com/whl/torch-1.5.0.html
python setup.py install or