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安装成功！
原文地址:https://blog.csdn.net/cyy0789/article/details/131137525
本文来自互联网用户投稿，该文观点仅代表作者本人，不代表本站立场。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。
如若转载，请注明出处：http://www.7code.cn/show_45230.html
如若内容造成侵权/违法违规/事实不符，请联系代码007邮箱：suwngjj01@126.com进行投诉反馈，一经查实，立即删除！
声明：本站所有文章，如无特殊说明或标注，均为本站原创发布。任何个人或组织，在未征得本站同意时，禁止复制、盗用、采集、发布本站内容到任何网站、书籍等各类媒体平台。如若本站内容侵犯了原著者的合法权益，可联系我们进行处理。
