【三步】最方便最简洁的TensorFlow 2.0 GPU版本安装教程

写在前面

1.你的电脑是有独立显卡的，支持GPU
2.你应该会下载安装 anaconda 并且应该会创建虚拟环境

版本问题

为了避免版本问题，请默认和我选择的版本匹配

开搞

第一步

win+R 打开CMD
激活你创建好的虚拟环境
activate tensorflow_gpu
不会创建虚拟环境
请看
https://zhuanlan.zhihu.com/p/94744929

输入conda install cudatoolkit=10.1
会提示你输入y回车就行

第二步

输入conda install cudnn=7.6

第三步

输入pip install tensorflow-gpu==2.1.0 -i https://pypi.tuna.tsinghua.edu.cn/simple

然后等待
一直等待

等待出现successfully

测试环节

输入python后
将以下代码分别输入

# tf2.0版本
import tensorflow.compat.v1 as tf

tf.disable_eager_execution()  # 可以用于从TensorFlow 1.x到2.x的复杂迁移项目的程序开头

hello = tf.constant('hello,tf')
sess = tf.Session()
print(sess.run(hello))

成功！

如果想看与pycharm关联起
你接着看

第一步

第二步

第三步

这里说明一下，你刚才创建好了的TensorFlow虚拟环境，你得晓得在哪里，你才能添加进去
一般情况
C:\Users\kingS\anaconda3\envs 你按着路径去找，一般在anaconda3下的envs


运行这段代码

# tf2.0版本
import tensorflow.compat.v1 as tf

tf.disable_eager_execution()  # 可以用于从TensorFlow 1.x到2.x的复杂迁移项目的程序开头

hello = tf.constant('hello,tf')
sess = tf.Session()
print(sess.run(hello))

最后hello,tf
收工