科技 > 人工智能 > 物联网

tensorflow 的2.15版 能解决安装tf的痛吗?

113人参与 2024-08-05 物联网

每次安装tensorflow最大的痛就是跟cuda的兼容问题。费半天劲安好了nvidia-driver、 cuda 、cudnn、tensorrt,再安装tensorflow,然后发现tensorflow还是不能启用gpu。

今天发现tensorflow已经发布2.15.0版了,更新介绍说这次可以通过pip一次把tensorflow和相应的cuda都安装好了。试了一下,果然省心多了。

官方安装指导:install tensorflow with pip (google.cn)

注意一定要在页面右上角把语言切换成英文,中文的内容(截止2023.12.04日)还没更新。

主要的安装步骤是:

python3 -m pip install --extra-index-url https://pypi.nvidia.com tensorrt-bindings==8.6.1 tensorrt-libs==8.6.1
python3 -m pip install -u tensorflow[and-cuda]

 安装过程会下载和安装很多个组件。我这边下载速度大概800k/s,安装用了一个小时。安装完成后用下面的语句验证:

python3 -c "import tensorflow as tf; print(tf.config.list_physical_devices('gpu'))"
(0)
打赏 微信扫一扫 微信扫一扫

您想发表意见!!点此发布评论

推荐阅读

2021数学建模国赛B题复盘详细解析

08-04

ClickHouse 与 Flink 整合:流处理与时间序列分析

08-06

云计算工程师个人简历(精选篇)

08-06

【图解物联网】第2章 物联网的架构

08-06

朴素贝叶斯在物联网数据分析中的实践

08-06

【物联网设备端开发】物联网设备上云提供开箱即用接入SDK

08-06

猜你喜欢

版权声明:本文内容由互联网用户贡献,该文观点仅代表作者本人。本站仅提供信息存储服务,不拥有所有权,不承担相关法律责任。 如发现本站有涉嫌抄袭侵权/违法违规的内容, 请发送邮件至 2386932994@qq.com 举报,一经查实将立刻删除。

发表评论