pytorch安装（2021）

最新推荐文章于 2025-05-19 16:50:36 发布

MInNrz

最新推荐文章于 2025-05-19 16:50:36 发布

阅读量5.2k

点赞数 18

CC 4.0 BY-SA版权

分类专栏： python

本文链接：https://blog.csdn.net/qq_40733911/article/details/113061874

python 专栏收录该内容

20 篇文章

订阅专栏

本文详细记录了如何在最新Anaconda和显卡驱动下安装PyTorch，包括检查显卡、创建并切换环境、更换镜像、安装步骤及注意事项，旨在帮助读者顺利规避安装难题。

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

Pytorch安装

安装这个东西花了我整整一天，踩了很多坑，记录一下，默认你已经安装了最新的anaconda和python，我把anaconda和显卡驱动都装成最新的了

可以使用如下命令更新
conda update python
conda update anaconda

第一：检查显卡，更新驱动

我的建议是首先更新驱动，全都按最新的东西来安装

https://www.nvidia.cn/Download/index.aspx?lang=cn#
在这里选择自己的显卡型号，下载安装即可，下载很快也很简单
建议大多数人都更新一下显卡驱动，避免不必要的麻烦

安装好后进入nvidia控制面板
鼠标右键可进入

点击组件找到自己CUDA的版本号

或者cmd输入

nvidia-smi
没有这条命令的需要添加环境变量（百度就有）

也可获得版本号
在这里插入图片描述

第二：创建pytorch环境

不同的项目需要不同的环境，可以处理不同版本的项目之间不兼容问题
进入anaconda prompt
在这里插入图片描述
输入

conda create -n mypytorch python=3.8

mypytorch是名字，3.8是python版本，都可以按自己需求改
然后按y，继续安装所需的包

输入

conda info --envs

可以看见自己的所有环境
在这里插入图片描述
接下来就是在mypytorch中安装pytorch包

第三：安装pytorch

进入pytorch官网

https://pytorch.org/

选择合适的版本（如果你的CUDA还是9.2，建议你返回第一步更新驱动），复制下面的命令（不要-c python）
在这里插入图片描述

在anaconda prompt中进入mypytorch环境输入

conda activate mypytorch

进入环境
在这里插入图片描述

重点来了
1.首先更换镜像（看第四点）
2.输入 conda install pytorch torchvision torchaudio cudatoolkit=11.0 （这段命令按自己的版本来）
注意后面没有-c pytorch，有的话记得删除
3.等待安装

第四：换镜像

注意所有操作都在mypytorch环境下
在这里插入图片描述

先清除原来镜像

conda config --remove-key channels

然后添加清华镜像
我的建议是直接改.condarc文件中的内容，我的内容如下，因为网上各路镜像都不一样，有的可以，有的不行，所以我直接全加进来了，建议你也这样
在这里插入图片描述

ssl_verify: true
show_channel_urls: true

channels:
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
  - http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/win-64/
  - http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/win-64/

在电脑中的.condarc文件中可以查看
在这里插入图片描述

最后就可以输入命令下载了

 conda install pytorch torchvision torchaudio cudatoolkit=11.0 （每个人的版本不一样）

下载过程相当的漫长，清华源也很慢，而且中途会http断开，包又很多个，多试几次，或者换个wifi可能会好点
在这里插入图片描述