1、确定自己显卡型号
我分到的是一台3060ti
2、查看当前显卡驱动版本
nvidia-smi
我的driver版本525.85.05
https://blog.csdn.net/liferecords/article/details/127528193
没装驱动的话自己装一下,看上图来说驱动最好版本高点,下面两种方法都ok
3、查看paddle官网安装
https://www.paddlepaddle.org.cn/install/quick?docurl=/documentation/docs/zh/install/pip/linux-pip.html
官网说安培架构的GPU和非安培架构的推荐cuda不一样,具体安培架构自己搜索,我记得3090这种贵的是安培架构,3060ti属于比较拉的了,装cuda10.2吧
python -m pip install paddlepaddle-gpu==2.4.2.post112 -f https://www.paddlepaddle.org.cn/whl/linux/mkl/avx/stable.html
4、安装cuda
https://developer.nvidia.com/cuda-toolkit-archive
5、安装cudnn
cudnn安装
总之,我目前的理解是paddle和pytorch不一样,pytorch可以通过一行命令安装,paddle需要提前下载安装好完整的cuda和cudnn才行,并且版本匹配,不然会报各种错误。比如:
W0305 16:35:06.338007 10608 gpu_resources.cc:61] Please NOTE: device: 0, GPU Compute Capability: 8.6
出现这种就是这几个的版本没匹配好导致的,需要重新安装对应版本的cuda,cudnn等。