当前位置:首页 » 《休闲阅读》 » 正文

GPU版本pytorch(Cuda12.1)清华源快速安装一步一步教!小白教学~

15 人参与  2024年04月06日 15:30  分类 : 《休闲阅读》  评论

点击全文阅读


GPU版本pytorch(Cuda12.1)清华源快速安装一步一步教!


作者为新电脑r9000p装pytorch时,没有注意自己的cuda版本是12.1(小伙伴自行百度下如何查看),装了旧版的torch结果import报错!找了半天原因才发现跟cuda版本不对应,要下载Cuda 11.8这个版本才可以,如下图:
在这里插入图片描述
在这里插入图片描述

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

上面是官方链接,不知道为什么我科学上网了下的还是非常慢,而且看网上有下好之后还有问题的,于是果断选择清华源!
下面是一步一步教程:
首先确保安装了anaconda,在anaconda prompt命令行窗口创建虚拟环境,我命名为 pytorch310(Python版本不要低于3.8)

conda create -n pytorch310 python==3.10

然后激活它:

conda activate pytorch310

设置清华源,加速安装:

pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple

开始指定版本安装:

pip install torch==2.0.0+cu118 torchvision==0.15.1+cu118 torchaudio==2.0.1+cu118 -f https://download.pytorch.org/whl/torch_stable.html

应该会非常快!我的下载速度平均在40M/s,两分钟就下好了!
然后输入conda list看一下有哪些库,这三个出来就没问题了!
在这里插入图片描述
不太放心,再试一下:
输入python,然后import torch,这个时候稍微等一下有点慢,然后torch.cuda.is_available(),如下图所示,出来True说明可以调用cuda!gpu版本pytorch安装成功!
在这里插入图片描述


点击全文阅读


本文链接:http://zhangshiyu.com/post/91254.html

<< 上一篇 下一篇 >>

  • 评论(0)
  • 赞助本站

◎欢迎参与讨论,请在这里发表您的看法、交流您的观点。

关于我们 | 我要投稿 | 免责申明

Copyright © 2020-2022 ZhangShiYu.com Rights Reserved.豫ICP备2022013469号-1