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