anaconda下安装torch_geometric

官方文档
安装的时候出了很多错误,最后还是按照官网文档才安装好了

文章目录

      • pytorch的版本
      • CUDA的版本
      • 安装包

pytorch的版本

确定pytorch的版本,最低1.4.0

python -c "import torch; print(torch.__version__)"
>>> 1.8.0

CUDA的版本

确定cuda的版本

python -c "import torch; print(torch.version.cuda)"
>>> 10.2

安装包

pip install torch-scatter -f https://pytorch-geometric.com/whl/torch-${TORCH}+${CUDA}.html
pip install torch-sparse -f https://pytorch-geometric.com/whl/torch-${TORCH}+${CUDA}.html
pip install torch-geometric

将其中${TORCH}${CUDA}替换为第一二步确定的torch和cuda版本。

pip install torch-scatter -f https://pytorch-geometric.com/whl/torch-1.8.0+cu102.html
pip install torch-sparse -f https://pytorch-geometric.com/whl/torch-1.8.0+cu102.html
pip install torch-geometric

你可能感兴趣的:(#,图神经网络,#,anaconda,环境,pytorch,深度学习,python)