之前装过pytorch,但是很多老的机器学习代码都是tensorflow,所以没办法,还要装个tensorflow。
在安装之前还要安装nvidia驱动还有cudn之类的,这些我已经在之前的篇章介绍过,就不在这细说了,可以直接传送过去看。
那么前面这些搞完,直接运行下面的命令:pip install --upgrade tensorflow-gpu
上面这行命令,会自动下载你的对应版本或者最新版本,现在下载应该是2.0版本的。
sess.run()无法运行
问题产生的原因:无法执行sess.run()的原因是tensorflow版本不同导致的,tensorflow版本2.0无法兼容版本1.0.
解决办法:tf.compat.v1.disable_eager_execution()
具体的测试tensorflow是否装好代码:
import tensorflow as tf
tf.compat.v1.disable_eager_execution() #保证sess.run()能够正常运行
hello = tf.constant('hello,tensorflow')
sess= tf.compat.v1.Session()#版本2.0的函数
print(sess.run(hello))
参考链接:
https://blog.csdn.net/weixin_38410551/article/details/103631977