nvidia驱动和pytorch版本不匹配问题

驱动和pytorch版本不匹配报错：
/home/anaconda3/envs/yolov5/lib/python3.7/site-packages/torch/cuda/init.py:52: UserWarning: CUDA initialization: The NVIDIA driver on your system is too old (found version 10010). 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. (Triggered internally at /pytorch/c10/cuda/CUDAFunctions.cpp:115.)
return torch._C._cuda_getDeviceCount() > 0

安装的驱动的430，pytorch是1.9.0.+cuda10.2
两种解决方法：升级驱动版本或降低pytorch版本。
升级驱动版本：
ubuntu-drivers devices
发现推荐的版本最高就是430了，用官网下载的470，安装.run。折腾了半天一直失败。最后放弃。
降低pytorch版本：
https://www.zhihu.com/question/309583980
430的驱动，最高支持的cuda toolkit版本的cuda 10.1。而我之前安装的torch1.9.0+cuda10.2，所以报错。
根据https://pytorch.org/get-started/previous-versions/的推荐，重新卸载并安装的pytorch，成功。

conda install pytorch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 cudatoolkit=9.2 -c pytorch