安装paddlepadddle-gpu的正确方式

正确安装paddlepadddle-gpu的方式

• 1.查看系统CUDA版本
• 2.参照飞桨官网快速pip安装

安装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