检查cuda安装_pytorch-geometric安装

本文指导如何检查CUDA安装并安装PyTorch-Geometric库,包括检查PyTorch版本,确认CUDA的成功安装及版本,以及一次性安装相关依赖,解决找不到模块的问题。示例给出的是CUDA10.1和torch1.5.0的安装命令。
摘要由CSDN通过智能技术生成
https://pytorch-geometric.readthedocs.io/en/latest/notes/installation.html​pytorch-geometric.readthedocs.io

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

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

PyTorch​pytorch.org
a1f0529a13494ff2cdfd86331e424b3f.png

检查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 &

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值