一、问题描述 使用gpu运行深度学习代码时,遇到这个错误。 二、错误原因: 与cuda驱动的版本有关。 三、解决办法: 1.打开命令行输入:nvidia-smi 2.查看CUDA版本 3.打开pytorch官网,下载对应CUDA版本的pytorch