注：下载cuda及cudnn安装教程，我是跟着下面文章安的：https://blog.csdn.net/jhsignal/article/details/111401628

第一个是电脑支持的最高cuda版本号，详见文章：https://blog.csdn.net/duoduohair/article/details/119761133。 第二个是pytorch默认支持的的几个版本，打开pytorch安装网址：https://pytorch.org/get-started/locally/，见下图：选择windows,conda,python会出来cuda10.2和cuda11.3为适合版本，当然也可以在下面代码中改成其他版本，我没改过。 如果下载错了，有三种方法：第一种，不用cuda了，直接用cpu，后期可能会出错，第二种，再安一个版本，两个版本之间不会影响，使用时需要换个环境，见文章：https://blog.csdn.net/