报错内容:
RuntimeError: The NVIDIA driver on your system is too old (found version 10000). 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 版本与 cuda 版本不匹配问题
解决:
# CUDA 10.0
pip install torch==1.4.0 torchvision==0.4.0 -f https://download.pytorch.org/whl/torch_stable.html