安装GPU版本的tensorflow就可以了pip install --upgrade https://storage.googleapis.com/tensorflow/windows/gpu/tensorflow_gpu-0.12.0rc0-cp35-cp35m-win_amd64.whl
还需要安装GPU加速包,下载 CUDA 和 cuDNN 的安装包了,注意版本号分别是 CUDA 8.0 和 cuDNN 5.1。
如果都安装成功,运行程序时会自动使用GPU计算
通过浏览器在线直接直接调用tensorflow环境,利用cuda编程查看GPU的型号信息