安装 Pytorch geometric

安装 Pytorch 的 geometric 库,主要内容参考官方文档

推荐安装方式为使用conda

conda install pyg -c pyg

另,网页给出了pytorch及其对应的cuda版本查询方法:

python -c "import torch; print(torch.__version__)"
>>> 1.11.0
python -c "import torch; print(torch.version.cuda)"
>>> 11.3

你可能感兴趣的:(pytorch,pytorch,深度学习)