补充:测试Tensorflow是否支持GPU的代码。
import tensorflow as tf
sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))
配环境这个事真的是常看常新,随着package的不断更新,之前配环境的经验都不能用了,所以新开一贴,记录一下最近配环境的曲折之路。
之前博客Windows10 GPU版Tensorflow配置教程+Anaconda3+Jupyter Notebook依然可以用。
#Windows10+GPU+Anaconda3+Pytorch+TensorFlow+Keras配置方法
最近由于人脸检测,接触到了pytorch框架,首先按照深度学习人脸识别系统 DFace移植WIN64环境详解(支持GPU)配置了一下。但是这位博主的经验也有点久远了,我按照他的来做并没有成功。摸索出来的成功配置步骤可以总结如下:
####1.环境信息
操作系统:Windows10
Anaconda3版本:4.2.0
CUDA版本:8.0
cuDNN版本:5.1
假设这些都已经配置好了,没有的话可以参考
Windows10 GPU版Tensorflow配置教程+Anaconda3+Jupyter Notebook
####2.准备配置文件
(1)environment-win64.yml
这是用来创建Anaconda环境的配置文件
用Notepad++打开,文件第一行可以修改自己想要的environment名称ÿ