GPU版本pytorch（Cuda12.1）清华源快速安装一步一步教！小白教学~

小曹同学努力了吗

已于 2024-09-02 14:28:49 修改

阅读量10w+

点赞数 174

文章标签： pytorch python 深度学习

于 2023-06-10 03:20:52 首次发布

本文链接：https://blog.csdn.net/cyy0789/article/details/131137525

版权

本文是针对使用CUDA12.1的用户安装GPU版PyTorch的教程。作者通过错误经历提醒读者注意CUDA版本匹配，提供了使用清华源加速安装PyTorch2.0.0+cu118、torchvision0.15.1+cu118和torchaudio2.0.1+cu118的步骤，包括创建Anaconda虚拟环境、设置清华源以及验证安装成功的测试方法。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

GPU版本pytorch（Cuda12.1）清华源快速安装一步一步教！

/===========================
写在头部: 尽作者所知，按照文章步骤，在去年（2023年）可以成功快速地安装GPU版pyTorch（可以达到你的网络带宽下载上限），目前陆续有同学反映按照所给步骤仍然是龟速安装（可能是哪一个软件版本和我当时没对应上）。鉴于本文所采用的步骤简单快速，有anaconda的同学可以快速试一下，如果不行可以看看评论区里，直接去镜像官网下载手动所需torch版本，这是最直接有效的办法，就是比较麻烦。
另：作者目前已经毕业工作，从事数字ic方向，欢迎有志之士一起交流。
感谢！
于2024年9月2日
===========================/
作者为新电脑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安装成功！