正确安装paddlepadddle-gpu的方式

安装paddlepaddle时,pip install paddlepaddle是直接安装的CPU版本,要安装GPU版本的话,就要注意适配的CUDA版本,安装GPU版本可参照官网教程,两步快速解决。

1.查看系统CUDA版本

nvcc -V

查看到系统cuda版本是11.2,如下所示:
请添加图片描述
注:为什么不用nvidia-smi查看,具体原因参考此博客

2.参照飞桨官网快速pip安装

官方网址:https://www.paddlepaddle.org.cn/
请添加图片描述
选择适配系统的cuda安装命令即可,这个跟python版本都适配,我用python3.7-3.9都安装成功了。
如下所示,我采用的安装命令为:

python -m pip install paddlepaddle-gpu==2.5.1.post112 -f https://www.paddlepaddle.org.cn/whl/linux/mkl/avx/stable.html
Logo

有“AI”的1024 = 2048,欢迎大家加入2048 AI社区

更多推荐