您的位置:首页 > 其它

TensorFlow - win7 - CPU/GPU版本安装

2017-08-02 13:39 537 查看

CPU requirement:

1. Anaconda
2. python 3.6
3. TensorFlow安装包


其中TensorFlow的安装包可以去下面网址下载对应的版本:

https://github.com/tensorflow/tensorflow

比如,我用的是:tensorflow-1.2.1-cp36-cp36m-win_amd64.whl,然后将下载的包放在D:\盘下面

ctrl + shift + cmd
,用超级管理员打开cmd

conda create -n your_env_name python=X.X(2.7、3.6等) anaconda
创建一个conda的虚拟环境

activate my_tensorflow
(这里my_tensorflow是我的虚拟环境) 激活这个虚拟环境

D:\
转到TensorFlow安装包所在的路径,我这里是D盘根目录

pip install tensorflow-1.2.1-cp36-cp36m-win_amd64.whl
直接安装,Done。

python
进入python 环境

import tensorflow
没报错,安装完成。好了,筒子们,可以用咯。。

conda install -n your_env_name your_package
当然你还可以用此命令在你的环境中安装python的各种包

deactivate
最后别忘了用这个命令关闭虚拟环境哦。

网上教程太复杂,还是自己的这个办法简单粗暴有效。。

GPU requirement:

一、安装cuda 9.1+VS2017

一路下一步即可,环境变量cuda会自动配好。

安装完后,在主目录搜索deviceQuery.exe,跑一跑测试下就好了(成功最下一句是:RESULT PASS)。

vs2017和cuda 9.1同时安装的,虽然cuda暂时提示找不到vs,但是不影响。

二、安装cudnn7.0

下载后解压,然后复制文件:

Copy the following files into the CUDA Toolkit directory.

a) Copy \cuda\bin\cudnn64_7.dll to C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\bin.

b) Copy \cuda\ include\cudnn.h to C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\include.

c) Copy \cuda\lib\x64\cudnn.lib to C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\lib\x64.

三、tensorflow安装

据说tf不支持cuda 9.1(因为官方的wheel是在cuda 9.0上编译的~)

当然有大神自己编译了一堆whl,方便不同版本使用。

https://github.com/fo40225/tensorflow-windows-wheel

找到对应cuda9.1版本,tf1.5的whl文件。

安装成功。
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: