【python】使用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版本

1. 查看pytorch版本

import torch
print(torch.__version__)

2. 查看cuda版本

nvidia-smi

3. 重装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.