pytorch安装

1.确定显卡型号

2.确定显卡算力：3.5

https://developer.nvidia.com/cuda-gpus

3.确定CUDA Runtime：

 https://zhuanlan.zhihu.com/p/544337083

4.确定显卡驱动版本：

（1）安装CUDA （如果电脑有则不需要安装）

https://www.nvidia.cn/Download/index.aspx?lang=cn
https://blog.youkuaiyun.com/qq_44703886/article/details/112859392

（2）查看自己的驱动 CUDA Driver Version --11.4

5.最终确定我们适用于：CUDA 11.0-11.4版本

CUDA　runtime　version版本＜＝CUDA　driver　version版本
CUDA　runtime　version版本　支持　显卡对应的算力

6.安装pytorch

（1）官网地址匹配下载命令

https://pytorch.org/

如果要安装以前的版本：ctrl+F搜索自己需要下载的版本，如11.3（根据标题5，要小于11.4，大于11.0）

（2）anaconda安装

依次输入：

conda create -n pytorch-XTD python=3.8

conda activate pytorch-XTD

conda install pytorch==1.12.1 torchvision==0.13.1 torchaudio==0.12.1 cudatoolkit=11.3 -c pytorch

（3）验证是否安装成功

依次输入：

python
import torch
torch.cuda.is_available()

如果结果显示：True ，则安装成功！