RuntimeError: cuDNN error: CUDNN_STATUS_NOT_INITIALIZED

cuda11运行pytorch报错

Traceback (most recent call last):
  File "train.py", line 220, in <module>
    main(args)
  File "train.py", line 89, in main
    train(model, args, train_loader, criterion, Hash_center, optimizer, epoch, two_loss_epoch)
  File "train.py", line 121, in train
    y = model(input)
  File "/root/miniconda3/envs/mocov3/lib/python3.7/site-packages/torch/nn/modules/module.py", line 889, in _call_impl
    result = self.forward(*input, **kwargs)
  File "/root/miniconda3/envs/mocov3/lib/python3.7/site-packages/torch/nn/parallel/data_parallel.py", line 167, in forward
    outputs = self.parallel_apply(replicas, inputs, kwargs)
  File "/root/miniconda3/envs/mocov3/lib/python3.7/site-packages/torch/nn/parallel/data_parallel.py", line 177, in parallel_apply
    return parallel_apply(replicas, inputs, kwargs, self.device_ids[:len(replicas)])
  File "/root/miniconda3/envs/mocov3/lib/python3.7/s