在服务器运行着很正常,撤掉服务器后,第一次运行没问题,再运行一次就显示了下面的错误:
RuntimeError: CUDA error: CUBLAS_STATUS_NOT_INITIALIZED when calling `cublasCreate(handle)`
搜了很多方法,猜想是运行内存不足。然后,除了pycharm,其余软件都关了,然后出现了标题的错误:
【pytorch】RuntimeError: cuDNN error: CUDNN_STATUS_INTERNAL_ERROR
所以更加肯定是内存问题,然后重启电脑,再运行,就好了。