记录自己安装pytorch出现的各种问题,希望对大家能有用
注:下载cuda及cudnn安装教程,我是跟着下面文章安的:https://blog.csdn.net/jhsignal/article/details/111401628
1、首先,下载cuda时一定要先看好要下载哪个版本,需要看以下几个地方:
第一个是电脑支持的最高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/zsc201825/article/details/91149550
第三种,卸载了重装(我是这么干的),见文章:https://blog.csdn.net/u013066730/a