BUG
AssertionError: The NVIDIA driver on your system is too old (found version 10010)
BUG分析
经过思考认为,应该是 Torch 和 Cuda 版本不匹配导致的,最后成功解决。解决思路如下
BUG解决
1、检查 Nvidia & Cuda 版本,使用 Linux 指令
nvidia-smi
2、判断torch版本是否支持
3、根据相关适配信息,选择安装版本;
4、目前来看明显是torch的问题
重新安装torch
pip install torch==1.5.0+cu101 torchvision==0.6.0+cu101 -f https://download.pytorch.org/whl/torch_stable.html
5、成功解决问题,亲测。