torch-geometric的安装步骤


torch-geometric:一个图神经网络的工具包。

零。安装步骤

官方链接:https://pytorch-geometric.readthedocs.io/en/latest/notes/installation.html
1,新建conda环境(可选)

conda create -n geo1 python=3.6
conda activate geo1
conda deactivate
nvcc --version	

10.0
nvcc 版本为10.0 ,所有想找torch 100的,但Geometric似乎只支持torch1.4和torch1.5。

2, 安装pytorch和对应驱动
pytorch官网

pip install torch==1.4.0+cu100-f https://download.pytorch.org/whl/torch_stable.html

注意,这个cu100非常重要,一不小心就白下载并且安装不了。

完成后查看版本

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

torch.version.cuda 10.0
torch.version 1.4.0+cu100

到这里,已经完成最重要的一步了,就是nvcc --versiontorch.version.cuda的版本均为10.0,这里要相同。后面的问题记录都是因为这两者不同引起的。

3,清空cache(最好这样)
因为Stored in directory: /root/.cache/pip,所以

rm -rf ~/.cache/pip

网上还有一种方法,

pip install --verbose -
  • 6
    点赞
  • 41
    收藏
    觉得还不错? 一键收藏
  • 13
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 13
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值