Pytorch报错:cuDNN error: CUDNN_STATUS_NOT_INITIALIZED与CUDA error: device-side assert triggered

解决方法

我尝试了网上的各种方法,很多都没有用,最后发现在代码的最前面添加一行语句可以解决报错。在此记录一下。

torch.backends.cudnn.enabled = False

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