使用grad-cam包过程中遇到的问题
grad-cam包
Crad-Cam的原理就不在这里展开,为了偷懒想要使用Pytorch中的grad-cam的类库,但是遇到了一些问题。
安装pytorch-grad-cam
在Pycharm终端进行安装
pip install grad-cam
问题1:The NVIDIA driver on your system is too old (found version 10010).
在安装之前我使用的torch版本是0.4.0,但是在安装grad-cam的过程中,由于grad-cam支持torch版本>=1.9.0.但建议是1.10.0,是比较稳定的。所以在安装过程中,torch进行了自动升版本。而我的cuda版本是10.1.在官方网站是找不到torch=1.10.0+cu101这样的搭配,但可以是torch=1.8.1+cu101.
如下操作查看自己pytorch版本和cuda版本
- 查看pytorch版本
import torch
print(torch.__version__)
- 查看cuda版本
nvidia-smi
- 重装pytorch+CUDA
pip3 install torch==1.8.1+cu101 torchvision==0.9.1+cu101 -f https://download.py torch.org/whl/cu101/torch_stable.html
综上所述,要使用grad-cam类库,必须满足torch>=1.10.0, cuda >=10.2否则会报错!!!
问题2: ModuleNotFoundError: No module named ‘torch.fx’
这个问题其实是上一个问题的一个延申,我自己的cuda版本是10.1,所以我就想使用torch 1.8.0+cu101的组合,但是报错了。出现这个报错的原因是,torch.fx这个模块要求torch版本大于等于1.9.0.但是cuda 10.1的版本适配的torch的最高版本是1.8.1
参考链接
链接: link.