Pytorch学习笔记：Pytorch版本与cuda版本不兼容

当我们辛辛苦苦改完代码，并且想利用GPU进行模型的训练时，可能会出现如下问题：

那么是什么原因造成的呢？在我们刚刚接触pytorch的时候可能不会考虑到pytorch版本与cuda版本匹配，所以装成了与我们电脑cuda版本不同的pytorch。那么接下来我们需要如下操作：

1.首先我们要先打开终端，查到自己的cuda版本，输入：nvidia-smi

可见，我的版本是cuda11.6

2.打开Start Locally | PyTorch，根据自己的版本选择

 我找了一下，目前最高的就是CUDA11.3的pytorch版本，所以选了这个。注意你的CUDA版本如果是10.几的，你这里面也只能选CUDA10.几。我的一开始下在的是这里面的CUDA10.2，而我的CUDA是11.6，就不可以用GPU加速。