写在前面
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
收工