Pytorch安装配置简洁版

1 准备NVIDIA显卡

1.1 查看显卡驱动和CUDA Driver

nvidia-smi

1.2 准备pytorch官网

一般不需要科学上网
Pytorch官网
如下所示，选择稳定版即可

如图NOTE中指出需要的python版本，下面分别是各种条件。本人CUDA的Driver是12.3，所以选择12<Platform<=12.3。这一方面的介绍可以观看B站一位叫做“我是土堆”的up视频。别忘了给人家点赞支持。
B站的土堆up视频，观看第25节视频

然后复制conda指令即可。

2 安装

2.1 创建新环境

这些环境的介绍在土堆视频里面也有。

在anaconda的prompt窗口输入这些创建新环境并进入即可。“pytorch1.0可以自定义名字”

conda create -n pytorch1.0 python=3.8

conda activate pytorch1.0

2.2 下载包

李沐老师这里回答了CUDA需不需要单独安装的问题，可以了解下。
忽视上面内容，这里介绍不需要安装CUDA的办法，下文参考一栏给出了CUDA的单独下载配置办法。

复制pytorch官网提供的conda指令

注：因为pytorch服务器在国外，一般下载都贼慢，本人亲测，挂了梯子也不行。所以建议换国内的源。找到condrac文件，一般在C:\Users\电脑用户名，这个地址下面。原本文件应该只有俩括号。建议修改后下载好pytorch再换回原来的那俩括号。这个办法是看了另一篇博客但忘了是谁的了。

{}

直接复制下面的内容到文件里。

channels:
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
  - defaults
show_channel_urls: true

然后保存再打开prompt窗口，将刚才pytorch官网给的conda指令输进去并回车执行。等待安装。
安装好之后开一个prompt命令行窗口

python

>>> import torch
>>> torch.cuda.is_available()
True
>>> print(torch.__version__)
1.12.0
>>> print(torch.version.cuda)
11.6
>>> print(torch.cuda.device_count())
1

依次输入这些绿箭头的指令看有无相似的输出。

参考

https://blog.csdn.net/qq_43596278/article/details/136305903点击跳转
这篇文章的评论说不需要再单独安装CUDA和cudatoolkit，cudnn。pytorch官网已经包括了

https://blog.csdn.net/YYDS_WV/article/details/137825313点击跳转
https://blog.csdn.net/weixin_43848614/article/details/117221384点击跳转
这两篇文章是完整的cuda配套的cudatooklkit和cudnn下载教程

另外还有一个踩坑版的pytorch安装教程，找不到网址了，大家如果看到了可以留意下。