安装torch以及版本对应问题

G鲲鹏展翅Y

已于 2024-03-10 19:51:39 修改

阅读量1.2k

点赞数 7

分类专栏： pytorchPyTorch （英语）文章标签： jupyter python pip

于 2024-03-10 10:10:02 首次发布

本文链接：https://blog.csdn.net/qq_44665922/article/details/136596381

版权

1 篇文章 0 订阅

订阅专栏

首先查看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

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

torch1.13.1对应的torchtext版本是0.14.1

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

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

关注