Windows7 64位 安装tensorflow

来源:互联网 发布:js导出excel 数字格式 编辑:程序博客网 时间:2024/05/22 03:40

上网查了一下windwos7 64位系统下安装tensorflow的一些资料,然后就开始安装了,安装过程还算顺利,但还是有几个需要注意的坑……

说明:
Windows7 64位系统,显卡支持gpu加速就直接安装了tensorflow-gpu的版本。安装之前先查一下,如果自己电脑的显卡不支持gpu加速的话,就可只安装tensorflow的cpu版本,无需安装CUDA和cuDNN。

安装环境

  • 操作系统:Win7 64位 sp1(没有sp1可能会安装失败,建议安装前升级到sp1)
  • python版本:python3.5.3,官网下载地址:https://www.python.org/downloads/windows/
  • CUDA版本:cuda_8.0.61_windows.exe,官网下载地址https://developer.nvidia.com/cuda-downloads
  • cuDNN版本:cudnn-8.0-windows7-x64-v5.1.zip,官网下载地址:https://developer.nvidia.com/cudnn
  • tensorflow版本:1.0.1,离线下载地址(cpu版):https://storage.googleapis.com/tensorflow/windows/cpu/tensorflow-1.0.1-cp35-cp35m-win_amd64.whl,离线下载地址(gpu版):https://storage.googleapis.com/tensorflow/windows/gpu/tensorflow_gpu-1.0.1-cp35-cp35m-win_amd64.whl

建议直接从以上网址下载这些各自对应的软件包,如果下载有问题,再从我的百度云网盘下载,下载地址:http://pan.baidu.com/s/1bo5GvWN,密码:4fln。
注意自己windwos操作系统的版本是否是win7 64位以及是否支持GPU加速再决定下载。

此处的坑:注意cuDNN的版本必须是5.1,使用5.0的会有问题(已踩过此坑,换了5.1就没问题了)

安装前的准备

1.确保你的系统已经安装python3.5 64位;
2.确保稳定的网络连接;
3.确保pip的版本>=8.1;
4.确认自己的显卡支持CUDA,如果不支持可以安装tensorflow的CPU版本(不需要安装CUDA和cuDNN)。

网上有的人说必须要安装VS,我的电脑以前安装过VS2010,然后好长时间没用,就给卸载了,卸载的干干净净。但还是安装成功了。不放心的可以先安装VS2010以后再进行安装。

安装CUDA

首先,可以去这个网址(https://developer.nvidia.com/cuda-gpus)查看自己的显卡是否支持。如果支持的话,然后去这个网址(https://developer.nvidia.com/cuda-downloads)下载自己windows版本的对应的exe安装包进行安装。

此处需要注意的坑:直接双击exe,一直默认下一步下一步结果安装失败。
后来,网上查资料,说是安装的时候选择自定义安装,然后不要勾选driver进行安装。试了一下,果然安装成功了。然后再双击exe再安装一次,这次安装只选择driver就可以了(又一大坑:我没有安装dirver就安装了tensorflow-gpu,结果安装完测试发现,python3.5的IDLE只要一运行tensorflow的session的run方法,python就挂了然后自动重启。这个坑我已帮各位提前踩了……如果发现安装dirver失败那就像我安装的一样,直接进行下一步吧,如果运行tensorflow失败的话,再把dirver安装上)。
安装CUDA的时间稍微有点儿长,而且安装dirver的时候屏幕可能会闪烁。
安装完成后打开cmd,命令行输入nvcc -V,看到版本信息则表示安装成功。
测试安装CUDA是否成功

安装cuDNN

下载地址:https://developer.nvidia.com/cudnn
下载的时候需要注册,然后填一个调查问卷,很简单。
下载完了会发现其实就是一个压缩包,将此压缩包解压到任何一个目录下都行,然后把目录添加到path环境变量。
例如解压到C盘,那么就把C:\cuda\bin添加到path环境变量。

此处注意,需要将cuda文件夹下的文件复制到相应的位置,如:

  • C:\cuda\bin\cudnn64_5.dll 复制到C:\Program Files\NVIDIA GPU Computing
    Toolkit\CUDA\v8.0\bin目录下;
  • C:\cuda\include\cudnn.h复制到C:\Program Files\NVIDIA GPU Computing
    Toolkit\CUDA\v8.0\include目录下
  • C:\cuda\lib\x64\cudnn.lib复制到C:\Program Files\NVIDIA GPU Computing
    Toolkit\CUDA\v8.0\lib\x64目录下。

安装tensorflow-gpu

安装tensorflow-gpu之前先要安装numpy。直接在cmd中输入pip install numpy进行安装可能会安装失败,不建议在线安装numpy。建议从http://www.lfd.uci.edu/~gohlke/pythonlibs/#numpy网站下载离线numpy的whl文件进行离线安装。下载的是numpy+mkl的whl文件(貌似直接pip install numpy安装后不包含mkl,可能会出问题)。
安装离线的numpy文件还有个坑,就是下载后死活安装不上,提示什么文件名之类的错误。我是用迅雷下载的,下载后文件名有乱码,安装失败。改了文件名以后还是安装失败。后来使用谷歌浏览器下载后,文件名不是乱码。打开cmd,进入whl文件所在的目录,然后输入pip install numpy‑1.11.3+mkl‑cp35‑cp35m‑win_amd64.whl安装成功。
tensorflow的安装,可以下载tensorflow的离线版,在不连接网络的情况下安装。但是建议在联网的情况下进行安装,因为可能会有相应的依赖库也需要进行安装(这个不太确定,如果在线安装失败,那么只能下载离线版安装了。我是在线安装,安装没问题)。
cmd下输入pip install tensorflow-gpu进行安装。网上有人说需要修改pip国内原,这个我没有修改,但也安装成功了。

测试

安装完成后,打开python3.5的IDLE,输入代码进行测试是否安装成功。
测试是否安装成功

如果能看到此结果,则说明安装成功!

参考博客:
1. 【TensorFlow】Windows10 64位下安装TensorFlow -
官方原生支持(http://blog.csdn.net/u010099080/article/details/53418159)
2. Windows7 64位安装TenSorFlow(http://blog.csdn.net/hongzhen91/article/details/62888660)
3. windwos安装tensorflow(http://blog.csdn.net/bixuange/article/details/53491635)

1 0
原创粉丝点击