安装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
相关推荐
鄃鳕14 小时前
python 字典 列表 类比c++【python】
c++·python
可触的未来,发芽的智生14 小时前
新奇特:黑猫警长的纳米世界,忆阻器与神经网络的智慧
javascript·人工智能·python·神经网络·架构
程序员三藏14 小时前
Jmeter接口测试与压力测试
自动化测试·软件测试·python·测试工具·jmeter·接口测试·压力测试
烛阴14 小时前
用 Python 揭秘 IP 地址背后的地理位置和信息
前端·python
大宝剑17014 小时前
python环境安装
开发语言·python
Element_南笙15 小时前
吴恩达新课程:Agentic AI(笔记2)
数据库·人工智能·笔记·python·深度学习·ui·自然语言处理
倔强青铜三15 小时前
苦练Python第69天:subprocess模块从入门到上瘾,手把手教你驯服系统命令!
人工智能·python·面试
倔强青铜三15 小时前
苦练 Python 第 68 天:并发狂飙!concurrent 模块让你 CPU 原地起飞
人工智能·python·面试
星期天要睡觉15 小时前
深度学习——循环神经网络(RNN)实战项目:基于PyTorch的文本情感分析
人工智能·python·rnn·深度学习·神经网络