解决Pytorch的cuDNN error: CUDNN_STATUS_NOT_INITIALIZED

目录

  • 1. 问题报错
  • 2. 可能原因
    • 2.1 GPU内存不足
    • 2.2 缓存问题
    • 2.3 CUDA和Pytorch版本不兼容
    • 2.4 CUDA和cuDNN版本不兼容
  • 3. 验证CUDA是否可用
  • 4. 参考


1. 问题报错

在使用GPU加速模型训练的过程中经常会遇到这样的错误:

RuntimeError: cuDNN error: CUDNN_STATUS_NOT_INITIALIZED

这个错误通常表示cuDNN库未能正确初始化。

2. 可能原因

2.1 GPU内存不足

在终端输入nvidia-smi查看GPU占用情况,如下图中的GPU 0几乎跑满,如果再使用该GPU运行其他占用内存较大的程序可能会报错。

>>> nvidia-smi

解决Pytorch的cuDNN error: CUDNN_STATUS_NOT_INITIALIZED_第1张图片

2.2 缓存问题

有时,cuDNN错误可能是由于缓存问题引起的。尝试清除缓存,然后重新运行代码。可以在Python代码中使用以下代码来清除缓存:

import

你可能感兴趣的:(Python,科研经验,Pytorch,pytorch,人工智能,python,GPU,CUDA,cuDNN)