pytorch运行错误CUDNN_STATUS_ARCH_MISMATCH

pytorch实现densenet的过程中,出现运行错误
但是运行简单的cnn以及resnet,均没有此问题
已邀请:

ptorch

赞同来自:

你使用的是什么版本的GPU?
CUDNN_STATUS_ARCH_MISMATCH 一般情况下表示你的GPU不支持CuDNN.
1、你可以测试设置torch.backends.cudnn.enabled为false,那样你的pytorch不会再使用CuDNN运行你的代码
2、还有就是pytorch默认选择gpu0,你可能选择了gpu1。pytorch与tensorflow不同

要回复问题请先登录注册