一、问题描述

今天在训练数据集的时候遇到如下报错：

 从报错上面可以看出是cuda出了问题，但是之前有一些需要用到cuda的程序也能正常运行，就觉得很奇怪。于是上网搜了一下具体错在哪。

网上有说法是pytorch和torchvision的版本没有对上号，也就是不兼容引发的报错。

二、问题解决

①查看电脑配置：

在终端输入下方命令查看配置：

nvidia-smi

② 到pytorch官网选择正确的版本：

 由于现在还没有CUDA11.8的版本，于是我选了最接近的CUDA11.6

③根据命令行下载：

 下载完毕，问题解决！