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等。

conda安装
cuda安装

Logo

腾讯云面向开发者汇聚海量精品云计算使用和开发经验,营造开放的云计算技术生态圈。

更多推荐