在conda环境下安装PyTorch+PyTorch Geometric(Pyg)

关于cuda和cudnn

不需要按网上教程去官网下载,也不需要手动在conda环境内安装。直接按PyTorch官网命令安装PyTorch即可,cuda和cudnn会自动按你选择的版本下载。

建立conda环境并激活

大家应该都会

conda create -n 你的环境名字 python=你想要的python版本
如:
conda create -n pytorch python=3.6

激活

conda activate 你的环境名字

安装PyTorch

官网链接:Previous PyTorch Versions | PyTorch

如果不需要特定版本,目前比较稳定的版本可选pytorch 1.8.0 +CUDA 10.2。激活环境后直接运行下面代码即可:

conda install pytorch==1.8.0 torchvision==0.9.0 torchaudio==0.8.0 cudatoolkit=10.2 -c pytorch

其他版本按官网指示来,需要改动几个数字

重点:不要安装最新版和带有".1"的版本,比如不要装 1.8.1(大坑!)。因为后面的Pytorch geometric (Pyg)兼容性不够好,会出各种问题。

更新:2022/4/26,上述命令不能用了,建议大家去官网使用最新版的指令

测试

python
import torch
print(torch.__version__)
print(torch.cuda.is_available())

返回你的PyTorch版本和Ture既是安装成功

安装Pyg

按CTRL+D退出python,如果上一步选择的PyTorch1.8.0版本(或1.8.0以上),此时直接运行下面代码即可,亲测1.8.0没问题,安装完成

conda install pyg -c pyg -c conda-forge

其他版本按官网指示Installation — pytorch_geometric 2.0.4 documentation

找到对应的pip代码运行即可

Tips

如果之前安装出现问题,建议新建一个环境,不在老环境里修改。新环境第一个就装PyTorch,会自动装需要的其他包。

用conda命令安装时它会检测你环境里缺少的包,然后自动给你一起安装。这很方便,但是有时会装一些不正确的包,导致一些冲突。所以一定要检查它要给你装哪些包,确认没问题在确认安装。

  • 24
    点赞
  • 38
    收藏
    觉得还不错? 一键收藏
  • 11
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值