由于项目需求,得用tensorflow完成,只能将mxnet的学习先放在一边,开始用tensorflow,废话不多说
首先安装anaconda+vs2015+cuda8.0+cudnn6.0
首先安装的anaconda自带3.7版本的python,但tensorflow并不支持,后来换成3.6.5版本的,发现同样不支持,真坑
后来在anaconda里面直接搜tensorflow-gpu,成功安装了tensorflow,而且自带3.6.2版本的python
注意:这里有两个版本的python,很容易出错,比如在用activate tensorflow-gpu激活python3.6.2后,用jupyter notebook并不能成功运行tensorflow,后来经过排查才知道jupyter在tensorflow-gpu环境下并没有,这样调用会自动调用3.6.5环境下的jupyter,所以不能运行。解决办法是在anaconda里面的tensorflow-gpu下下载一个jupyter,这样就可以了
至此,tensorflow-gpu能够成功运行了
在pycharm运行会出现如下警告:
虽然对结果没影响,但大片红色区看着碍眼睛,加上这两句就可以了
import os
os.environ['TF_CPP_MIN_LOG_LEVEL']='2'