0
点赞
收藏
分享

微信扫一扫

【PyTorch 08】如果要手动安装对应的包

何以至千里 2023-09-14 阅读 40

例如有时候我们要下载 PyG ,但是需要手动下载,需要进行以下步骤:
网站链接:https://data.pyg.org/whl/

首先查看当前安装好的Pytorch版本和对应的cuda版本

1. pip list:查看torch版本

2. torch.version.cuda:查看cuda版本

就可以找到下面这种形式

cp代表python版本,本机是3.10的,点击下载

 其他几个类似。

接下来使用下面这句话安装工具

pip install torch_geometric

举报

相关推荐

0 条评论