记录一下在m1pro和win10上安装torch-geometric

最新推荐文章于 2024-04-24 22:56:58 发布

黑旋风李白Lexie

最新推荐文章于 2024-04-24 22:56:58 发布

阅读量98

点赞数 2

文章标签： pytorch 深度学习

本文链接：https://blog.csdn.net/weixin_42632778/article/details/134105593

版权

m1pro

参考了这个issue的一个回答https://github.com/rusty1s/pytorch_scatter/issues/241

这个老哥是M1 Max + MacOS 13.0.1，我的设备是M1 Pro + MaxOS 12.3.1，亲测有用

conda create -n pyg-m1 python=3.10

conda activate pyg-m1

pip install --no-cache-dir torch==1.13.0 torchvision torchaudio
pip install git+https://github.com/rusty1s/pytorch_sparse.git
pip install git+https://github.com/rusty1s/pytorch_scatter.git
pip install git+https://github.com/rusty1s/pytorch_cluster.git

pip --no-cache-dir install torch-geometric

win10

一样的哈都是从头创建一个环境，依次安装torch和torch-geometric需要的依赖

从头来的好处就是不用管之前装没装torch，装了啥版本的，就这一套完事儿

conda create -n pyg python=3.8 -y

conda activate pyg

pip install torch_scatter torch_sparse torch_cluster torch_spline_conv -f https://data.pyg.org/whl/torch-2.1.0+cpu.html

pip install torch_geometric