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