RuntimeError: module 'torch.backends.cudnn' has no attribute 'allow_tf32'

[xuguozhi]

------------------------------------------------------------
sys.platform: linux
Python: 3.7.16 (default, Jan 17 2023, 22:20:44) [GCC 11.2.0]
CUDA available: True
CUDA_HOME: /usr/local/cuda
NVCC: Cuda compilation tools, release 10.0, V10.0.130
GPU 0: Tesla V100-SXM2-32GB
GCC: gcc (Debian 6.3.0-18+deb9u1) 6.3.0 20170516
PyTorch: 1.6.0
PyTorch compiling details: PyTorch built with:
  - GCC 7.3
  - C++ Version: 201402
  - Intel(R) oneAPI Math Kernel Library Version 2021.4-Product Build 20210904 for Intel(R) 64 architecture applications
  - Intel(R) MKL-DNN v1.5.0 (Git Hash e2ac1fac44c5078ca927cb9b90e1b3066a0b2ed0)
  - OpenMP 201511 (a.k.a. OpenMP 4.5)
  - NNPACK is enabled
  - CPU capability usage: AVX2
  - CUDA Runtime 10.1
  - NVCC architecture flags: -gencode;arch=compute_37,code=sm_37;-gencode;arch=compute_50,code=sm_50;-gencode;arch=compute_60,code=sm_60;-gencode;arch=compute_61,code=sm_61;-gencode;arch=compute_70,code=sm_70;-gencode;arch=compute_75,code=sm_75;-gencode;arch=compute_37,code=compute_37
  - CuDNN 7.6.3
  - Magma 2.5.2
  - TorchVision: 0.7.0
OpenCV: 4.7.0
MMCV: 1.4.2
MMGen: 0.7.3+67e7b17
MMCV Compiler: GCC 7.3
MMCV CUDA Compiler: 10.1

here is the error message:

Traceback (most recent call last):
  File "tools/train.py", line 161, in <module>
    main()
  File "tools/train.py", line 157, in main
    meta=meta)
  File "/home/agileGan/MMGEN-FaceStylor/mmgeneration/mmgen/apis/train.py", line 207, in train_model
    runner.run(data_loaders, cfg.workflow, cfg.total_iters)
  File "/home/agileGan/MMGEN-FaceStylor/mmgeneration/mmgen/core/runners/dynamic_iterbased_runner.py", line 285, in run
    iter_runner(iter_loaders[i], **kwargs)
  File "/home/agileGan/MMGEN-FaceStylor/mmgeneration/mmgen/core/runners/dynamic_iterbased_runner.py", line 215, in train
    outputs = self.model.train_step(data_batch, self.optimizer, **kwargs)
  File "/home/agileGan/MMGEN-FaceStylor/mmgeneration/mmgen/core/ddp_wrapper.py", line 123, in train_step
    output = self.module.train_step(*inputs[0], **kwargs[0])
  File "/home/agileGan/MMGEN-FaceStylor/agilegan/transfer.py", line 414, in train_step
    loss_gen, log_vars_g, source_results = self._get_gen_loss(data_dict_)
  File "/home/agileGan/MMGEN-FaceStylor/agilegan/transfer.py", line 121, in _get_gen_loss
    loss_ = loss_module(outputs_dict)
  File "/root/anaconda3/envs/facestylor2/lib/python3.7/site-packages/torch/nn/modules/module.py", line 722, in _call_impl
    result = self.forward(*input, **kwargs)
  File "/home/agileGan/MMGEN-FaceStylor/mmgeneration/mmgen/models/losses/gen_auxiliary_loss.py", line 265, in forward
    **kwargs)
  File "/home/agileGan/MMGEN-FaceStylor/mmgeneration/mmgen/models/losses/gen_auxiliary_loss.py", line 108, in gen_path_regularizer
    only_inputs=True)[0]
  File "/root/anaconda3/envs/facestylor2/lib/python3.7/site-packages/torch/autograd/__init__.py", line 190, in grad
    return Variable._execution_engine.run_backward(outputs, grad_outputs, retain_graph, create_graph,inputs, allow_unused)
RuntimeError: module 'torch.backends.cudnn' has no attribute 'allow_tf32'