The NVIDIA driver on your system is too old (found version 9020).解决方法

原本的环境是cuda9.0,pytorch 1.1.0版本的,在进行centerface训练模型的过程中,遇到以下报错:

AssertionError:
The NVIDIA driver on your system is too old (found version 9020).
Please update your GPU driver by downloading and installing a new
version from the URL: http://www.nvidia.com/Download/index.aspx
Alternatively, go to: https://pytorch.org to install
a PyTorch version that has been compiled with your version
of the CUDA driver.

查阅了网上的资料,其实是pytorch版本过新的问题,使用命令conda install pytorch=1.0.0 torchvision  安装1.0.0版的pytorch,

问题解决。

猜你喜欢

转载自www.cnblogs.com/xyzluck/p/12970331.html