安装Tensorflow-GPU

版本:win7 + python3.5 + tensorflow-gpu1.4.0 + 1080Ti + cuda8.0GA1 + cudnn6

1.python安装

anaconda3.4.2 对应python3.5,

https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/

会出现闪退问题,解决办法--->升级,管理员身份运行prompt,键入 

conda update anaconda-navigator

2.创建一个 tensorflow-GPU 环境,键入

conda create -n tensorflow python=3.5

3.安装TensorFlow

进入环境,键入

activate tensorflow

安装tensorflow  键入 

pip install tensorflow-gpu==1.4.0

4.会出现的问题:“No Model Named object_detection”,解决办法--->写一个.pth文件到该文件夹下D:\Anaconda\envs\tensorflow\Lib\site-packages,命名为tensorflow_model.pth,内容如下:

D:\0TFWin\models\research
D:\0TFWin\models\research\slim
D:\0TFWin\models\research\object_detection
***(非必选)安装gpu-z监控

需要安装的包有很多,可以直接在anacond中安装,用清华的镜像速度快,也可用pip安装,前提是activate tensorflow,主要的包有 matplotlib  lxml   pillow   numpy 

***cudnn下载的几个文件拷贝到 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0 下,此外显卡、显卡驱动、DUDA加速、cudnn,这4者的版本必须得一致,否则会出现错误。

猜你喜欢

转载自blog.csdn.net/jh0lmes/article/details/80148831