首先要知道自己用tensorflow搭建的神经网络是用cpu训练还是gpu训练。
以下是windows安装tensorflow cpu的步骤:
1.打开anaconda prompt:
2.此时处于base环境下,输入conda env list,查看有哪些虚拟环境:
3.接下来创建虚拟环境,输入conda create -n 虚拟环境名称 python=python版本,示例:conda create -n tf_cpu python=3.7
中途需要输入y
4.输入activate tf_cpu切换到刚创建的虚拟环境:
5.最后输入pip install tensorflow -i http://pypi.douban.com/simple --trusted-host pypi.douban.com进行安装,切记此处不要用conda安装,不然可能会造成在后续导包不成功:
安装后进行验证,进入python解释器,输入import tensorflow as tf
print(tf.version),出现如下界面:
至此windows安装tensorflow cpu完毕!