找到报错的位置,放宽if的条件
if any(torch.__version__.startswith(x) for x in ['1.7.', '1.8.', '1.9', '1.10']):
return True
warnings.warn(f'conv2d_gradfix not supported on PyTorch {torch.__version__}. Falling back to torch.nn.functional.conv2d().')
把自己的torch version版本放到for x in [‘1.7.’, ‘1.8.’, ‘1.9’, ‘1.10’])里面。