在用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-06-29 14:40:37 发布