当前位置:   article > 正文

在conda环境下安装PyTorch+PyTorch Geometric(Pyg)_conda install -y pyg -c pyg 失败

conda install -y pyg -c pyg 失败

关于cuda和cudnn

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

建立conda环境并激活

大家应该都会

  1. conda create -n 你的环境名字 python=你想要的python版本
  2. 如:
  3. 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,上述命令不能用了,建议大家去官网使用最新版的指令

测试

  1. python
  2. import torch
  3. print(torch.__version__)
  4. 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命令安装时它会检测你环境里缺少的包,然后自动给你一起安装。这很方便,但是有时会装一些不正确的包,导致一些冲突。所以一定要检查它要给你装哪些包,确认没问题在确认安装。

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/天景科技苑/article/detail/860459
推荐阅读
相关标签
  

闽ICP备14008679号