原本的環境是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,
問題解決。