参考博主文章
不同的是,遇到RuntimeError: Error building extension '_prroi_pooling'报错,没有重新下载torch
因为本来有cuda10.2版本,在'usr/local/'有cuda,cuda-10.2,cuda-11.2,因此改了配置文件bashrc
Summary on deep learning framework --- PyTorch - AHU-WangXiao - 博客园
参考上面链接解决的。
终端输入:export CUDA_HOME='/usr/local/cuda'
另一个报错:AttributeError: module ‘torch‘ has no attribute ‘irfft
【Pytorch】torch_dct中解决AttributeError: module ‘torch‘ has no attribute ‘irfft‘_宇宙爆肝锦标赛冠军的博客-CSDN博客
旧版pytorch中torch.rfft和irfft在新版本中的对应 - 知乎
尝试很多改代码的方法,技不如人
卸载pytorch,重新安装
原来是torch1.8.0+cu111,torchvision0.9.0
换成了torch1.7.1+cu110,torchvision0.8.2