安装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
相关推荐
云泽野3 小时前
【Java|集合类】list遍历的6种方式
java·python·list
IMPYLH5 小时前
Python 的内置函数 reversed
笔记·python
小赖同学啊7 小时前
物联网数据安全区块链服务
开发语言·python·区块链
码荼7 小时前
学习开发之hashmap
java·python·学习·哈希算法·个人开发·小白学开发·不花钱不花时间crud
小陈phd8 小时前
李宏毅机器学习笔记——梯度下降法
人工智能·python·机器学习
kk爱闹8 小时前
【挑战14天学完python和pytorch】- day01
android·pytorch·python
Blossom.1189 小时前
机器学习在智能建筑中的应用:能源管理与环境优化
人工智能·python·深度学习·神经网络·机器学习·机器人·sklearn
亚力山大抵9 小时前
实验六-使用PyMySQL数据存储的Flask登录系统-实验七-集成Flask-SocketIO的实时通信系统
后端·python·flask
showyoui9 小时前
Python 闭包(Closure)实战总结
开发语言·python
amazinging10 小时前
北京-4年功能测试2年空窗-报培训班学测开-第四十一天
python·学习·appium