如何根据CUDA版本安装对应的pytorch？

在官网中找到对应的安装命令

1. 进入pytorch官网，找到下面截图
   
2. 根据自己电脑的情况选择对应选项，如果电脑只有CPU，没有Nvidia显卡，Compute Platform那一项只能选CPU，如果有显卡，则需要查清楚自己电脑安装的CUDA版本。
3. 将 Run this Command中的命令复制到cmd中（cmd要以管理员身份运行，否则会报错）
4. 运行安装pytorch

如何查看自己电脑的CUDA版本？

1. 以管理员身份打开cmd，在桌面 window 标志旁边一般有个搜索框，输入cmd，有个应用叫命令提示符，右键以管理员身份运行。在Linux中就是terminal。
2. 输入nvcc -V，查看自己电脑的cuda版本，如下面截图显示的CUDA版本是11.5，Compute Platform那一项选CUDA11.3是可以的。
   

来源：我才是少尉