pytorch-geometrican安装

pytorch-geometrican安装

  • 1.安装pytorch,若已安装检查pytorch安装版本
  • 2.检查CUDA是否安装成功及其版本
  • 3.安装pytorch-geometric及其需要packages
  • 4.检查是否安装成功

仅供参考,建议根据官网安装

1.安装pytorch,若已安装检查pytorch安装版本

pytorch安装
检查pytorch安装版本

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

2.检查CUDA是否安装成功及其版本

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

3.安装pytorch-geometric及其需要packages

${CUDA}可以用cpu,cu92,cu100或者cu101代替。本步与官网有些不同,是一次性安装相关packages,可以解决安装成功后出现no module ‘pytorch-geometric’ 情况。

	pip install --no-cache-dir torch-geometric torch-sparse==latest+${CUDA} torch-spline-conv==latest+${CUDA} torch-scatter==latest+${CUDA} torch-cluster==latest+${CUDA} torch-spline-conv==latest+${CUDA} -f https://pytorch-geometric.com/whl/torch-version.html

示例:

	pip install --no-cache-dir  torch-geometric   torch-sparse==latest+cu101 torch-spline-conv==latest+cu101  torch-scatter==latest+cu101   torch-cluster==latest+cu101   torch-spline-conv==latest+cu101 -f https://pytorch-geometric.com/whl/torch-1.5.0.html

其中,以上安装语句是以CUDA10.1版本torch1.5.0为例,–no-cache-dir可加可不加。如果出现路径问题,可参考官网设置CUDA的路径。

4.检查是否安装成功

	python -c "import torch;import torch_geometric;import torch_sparse; print(torch.__version__,torch_geometric.__version__,torch_sparse.__version__)"

你可能感兴趣的:(图神经网络GNN)