Python报错：cuDNN error: CUDNN_STATUS_NOT_INITIALIZED 和 CUDA_LAUNCH_BLOCKING=1.

model [CycleGANModel] was created ---------- Networks initialized ------------- [Network G_A] Total number of parameters : 11.378 M [Network G_B] Total number of parameters : 11.378 M [Network D_A] Total number of parameters : 2.765 M [Network D_B] Total number of parameters : 2.765 M ----------------------------------------------- Setting up a new session... create web directory ./checkpoints\horse2zebra_cyclegan\web... D:\N 1\anaconda\Lib\site-packages\torch\optim\lr_scheduler.py:227: UserWarning: Detected call of `lr_scheduler.step()` before `optimizer.step()`. In PyTorch 1.1.0 and later, you should call them in the opposite order: `optimizer.step()` before `lr_scheduler.step()`. Failure to do this will result in PyTorch skipping the first value of the learning rate schedule. See more details at https://pytorch.org/docs/stable/optim.html#how-to-adjust-learning-rate warnings.warn( learning rate 0.0002000 -> 0.0002000 Traceback (most recent call last): File "E:\pytorch-CycleGAN-and-pix2pix-master\pytorch-CycleGAN-and-pix2pix-master\train.py", line 52, in <module> model.optimize_parameters() # calculate loss functions, get gradients, update network weights ^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "E:\pytorch-CycleGAN-and-pix2pix-master\pytorch-CycleGAN-and-pix2pix-master\models\cycle_gan_model.py", line 187, in optimize_parameters self.backward_G() # calculate gradients for G_A and G_B ^^^^^^^^^^^^^^^^^ File "E:\pytorch-CycleGAN-and-pix2pix-master\pytorch-CycleGAN-and-pix2pix-master\models\cycle_gan_model.py", line 178, in backward_G self.loss_G.backward() File "D:\N 1\anaconda\Lib\site-packages\torch\_tensor.py", line 626, in backward torch.autograd.backward( File "D:\N 1\anaconda\Lib\site-packages\torch\autograd\__init__.py", line 347, in backward _engine_run_backward( File "D:\N 1\anaconda\Lib\site-packages\torch\autograd\graph.py", line 823, in _engine_run_backward return Variable._execution_engine.run_backward( # Calls into the C++ engine to run the backward pass ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ RuntimeError: cuDNN error: CUDNN_STATUS_INTERNAL_ERROR_HOST_ALLOCATION_FAILED

yolo RuntimeError: cuDNN error: CUDNN_STATUS_NOT_INITIALIZED