当前位置:   article > 正文

安装torch以及版本对应问题_torch官对应版本

torch官对应版本

首先查看cuda版本:win+R 输入:nvidia -smi
在这里插入图片描述
我的cuda版本12.2,安装的torch版本要小于12.2

我的pip/conda源都改成清华源了,torch2.0以上的版本截止到2024年3月10日也没有。
pytorch官网:https://pytorch.org/
寻找匹配的版本命令,一个不行就多换几个,下面这条命令在我的电脑上没报错。

pip install torch==1.13.1+cu116 torchvision==0.14.1+cu116 torchaudio==0.13.1 --extra-index-url https://download.pytorch.org/whl/cu116
  • 1

torch torchvision torchaudio 这三个是一起下载的,还要安装torchtext,寻找torchtext匹配版本

torch1.13.1对应的torchtext版本是0.14.1

**注意:版本型号一定要匹配,否则会给日后实验埋雷,**不要问我为什么这么说。

不要更改和torch版本不匹配的torchtext,某则torch自动更新,torch torchvision torchaudio torchtext的版本全乱套了

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

闽ICP备14008679号