Torch not compiled with CUDA enabled

训练的数据集时出现的错误:

我们在安装系统时都会安装驱动,那么驱动有什么作用呢?

(来自驱动的作用_weixin_33676492的博客-CSDN博客

 驱动程序,全称为“设备驱动程序”, 是一种可以使计算机和设备通信的特殊程序,可以说相当于硬件的接口,操作系统只有通过这个接口,才能控制硬件设备的工作,假如某设备的驱动程序未能正确安装,便不能正常工作。

如何查看驱动是否安装?

cmd下:nvidia -smi

Torch not compiled with CUDA enabled_第1张图片

 如果没安装驱动,就不会显示信息。此时需要先安装驱动:
步骤: 官网:https://www.nvidia.cn/geforce/drivers/ -> 输入电脑相关配置 -> 选择:nvidia studio driver-> 获取下载, 即可。

如何确定Cudnn的版本?

Cudnn的版本就是根据CUDA的版本确定,

查看CUDA版本:nvcc -V

Torch not compiled with CUDA enabled_第2张图片

看了一下应该是cuda10.0版本,cudnn适配的cuda10.2版本

 于是卸载cuda10.0

 Torch not compiled with CUDA enabled_第3张图片

 重新下载cuda10.2

CUDA Toolkit 10.2 Download | NVIDIA Developer

使用迅雷下载会更快一点

Torch not compiled with CUDA enabled_第4张图片

 

运行cuda10.2,解压

Torch not compiled with CUDA enabled_第5张图片

 NVIDIA安装程序

Torch not compiled with CUDA enabled_第6张图片

 最后安装完成.

你可能感兴趣的:(Pytorch安装配置,python,深度学习)