安装paddlepaddle时,pip install paddlepaddle
是直接安装的CPU版本,要安装GPU版本的话,就要注意适配的CUDA版本,安装GPU版本可参照官网教程,两步快速解决。
1.查看系统CUDA版本
nvcc -V
查看到系统cuda版本是11.2,如下所示:
注:为什么不用nvidia-smi查看,具体原因参考此博客文章来源:https://www.toymoban.com/news/detail-658108.html
2.参照飞桨官网快速pip安装
官方网址:https://www.paddlepaddle.org.cn/
选择适配系统的cuda安装命令即可,这个跟python版本都适配,我用python3.7-3.9都安装成功了。
如下所示,我采用的安装命令为:文章来源地址https://www.toymoban.com/news/detail-658108.html
python -m pip install paddlepaddle-gpu==2.5.1.post112 -f https://www.paddlepaddle.org.cn/whl/linux/mkl/avx/stable.html
到了这里,关于安装paddlepadddle-gpu的正确方式的文章就介绍完了。如果您还想了解更多内容,请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章,希望大家以后多多支持TOY模板网!