安装paddlepadddle-gpu的正确方式

正确安装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
相关推荐
databook8 小时前
Manim实现脉冲闪烁特效
后端·python·动效
程序设计实验室8 小时前
2025年了,在 Django 之外,Python Web 框架还能怎么选?
python
倔强青铜三10 小时前
苦练Python第46天:文件写入与上下文管理器
人工智能·python·面试
用户25191624271113 小时前
Python之语言特点
python
刘立军13 小时前
使用pyHugeGraph查询HugeGraph图数据
python·graphql
数据智能老司机17 小时前
精通 Python 设计模式——创建型设计模式
python·设计模式·架构
数据智能老司机18 小时前
精通 Python 设计模式——SOLID 原则
python·设计模式·架构
c8i19 小时前
django中的FBV 和 CBV
python·django
c8i19 小时前
python中的闭包和装饰器
python
这里有鱼汤1 天前
小白必看:QMT里的miniQMT入门教程
后端·python