cuDNN error: CUDNN_STATUS_INTERNAL_ERROR 问题解决

训练过程中出现

 

解决方法:

1.

有老哥说

os.environ['CUDA_ENABLE_DEVICES'] = '0'
但是该方法在Pytorch代码并不适用,正确的操作方法是:torch.cuda.set_device(0)  #试了一下,没用

 

2.解决办法:关闭程序,重启电脑,再运行,好像也没用

 

应该是和cuda版本有关

 

 

 

你可能感兴趣的:(深度学习,人工智能)