【Pytorch】解决cuDNN error: CUDNN_STATUS_NOT_INITIALIZED
文章目录
问题描述
在git上下载了相关的代码准备在服务器上跑一下,然后发现在之前配好的conda虚拟环境下跑不了,版本太高了,所以就重新配了一个虚拟环境,结果遇到了以下问题
RuntimeError cuDNN error: CUDNN_STATUS_NOT_INITIALIZED
报这个错误的时候,不是代码的问题,而是环境的问题,所以首先我们来排查一下自己的虚拟环境
解决思路
1、验证cuda是否可用
conda activate your_env_name
python
import torch
print(torch.cuda.is_available(