安装tensorflow-gpu

最新推荐文章于 2021-01-30 11:02:41 发布

张zzz

最新推荐文章于 2021-01-30 11:02:41 发布

阅读量189

点赞数

CC 4.0 BY-SA版权

分类专栏：技术文章标签： tensorflow-gpu

本文链接：https://blog.csdn.net/weixin_41673190/article/details/92762733

技术专栏收录该内容

12 篇文章

订阅专栏

博客介绍了利用TensorFlow-GPU的方法。一是用pip直接安装，但过程慢且易失败；二是使用pypi源安装。还给出了测试方法，通过导入TensorFlow并查看版本来验证。同时提到可能存在安装版本与cuda不一致及后续更新匹配问题。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

在用tensorflow的时候，可能常常用到gpu，下面将介绍如何利用tensorflow-gpu。
有两种方法：（1）：pip install tensorflow-gpu版本，但是安装过程贼慢，甚至会安装失败。个人不建议使用。
（2）使用pypi源安装：例如：pip install -i https://pypi.tuna.tsinghua.edu,cn/simple tensorflow-gpu版本，或者pip install tf-nightly2.0-preview -i http://pypi.douban.com/simple --trusted-host pypi.douban.com
下面就要进行测试：
1 python
2 import tensorflow as tf
#查看tensorflow版本
3 print(tf.version)
这样算成功了，但是可能安装版本与cuda不一致，下次更新如何匹配。