在用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不一致,下次更新如何匹配。
安装tensorflow-gpu
最新推荐文章于 2021-01-30 11:02:41 发布
博客介绍了利用TensorFlow-GPU的方法。一是用pip直接安装,但过程慢且易失败;二是使用pypi源安装。还给出了测试方法,通过导入TensorFlow并查看版本来验证。同时提到可能存在安装版本与cuda不一致及后续更新匹配问题。
部署运行你感兴趣的模型镜像
您可能感兴趣的与本文相关的镜像
PyTorch 2.5
PyTorch
Cuda
PyTorch 是一个开源的 Python 机器学习库,基于 Torch 库,底层由 C++ 实现,应用于人工智能领域,如计算机视觉和自然语言处理
7531

被折叠的 条评论
为什么被折叠?



