环境监测:
import torch
print(torch.__version__) # 查看pytorch安装的版本号
print(torch.cuda.is_available()) # 查看cuda是否可用。True为可用,即是gpu版本pytorch
print(torch.cuda.get_device_name(0)) # 返回GPU型号
print(torch.cuda.device_count()) # 返回可以用的cuda(GPU)数量,0代表一个
print(torch.version.cuda) # 查看cuda的版本
下载地址:
https://pytorch-geometric.com/whl/
选择版本
安装依赖
进入cmd.跳转到指定目录,然后依次进行安装,截图如下:
安装 torch_geometric
pip install torch_geometric -i https://pypi.tuna.tsinghua.edu.cn/simple
这里使用了清华的镜像
测试
在pycharm里面引入torch_geometric 相关的代码。测试不报错