import torch
import tensorflow as tf
if __name__ == "__main__":
print('cuDNN version: ', torch.backends.cudnn.version()) # Corresponding cuDNN version
运行以上代码,出现以下错误提示。
显示错误为cuDNN version incompatibility: PyTorch was compiled against (8, 5, 0) but found runtime version (8, 2, 2)。
尝试查看cuDNN版本。
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
运行该指令没有结果,右键选择运行指令,可以看到cudnn.h文件内容,内含license信息,没有版本信息。
经过查询,cudnn版本信息目前存储于cudnn_version.h中。
cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2
cuDNN版本为(8, 0, 5)。跟cuDNN version incompatibility错误:PyTorch was compiled against (8, 5, 0) but found runtime version (8, 2, 2) 中,提到的版本号一致。