Tensorflow-GPU在window环境下安装历程

阅读更多

    Tensorflow是Google在2015年11月9日宣布开源的第二代机器学习系统,支持python和C++,支持CNN、RNN和LSTM等算法,这里python IDE推荐pyChorm,功能强大,安装库也很方便,简单的tensorflow的默认cpu版本可以直接pip安装运行没有问题,但是速度非常慢,所以改为使用tensorflow-gpu,速度能提升很多。

     首先查询电脑是否支持gpu,确定支持后,先下载NVIDIR的CUDA和CUDNN(可以在英伟达官网下载),CUDA是图像处理支持,CUDNN是链接tensorflow和CUDA必要的插件,这里CUDA必须下载8.0版本,9.0版本暂时不支持。这里对版本的对应非常严格,之前下载哈工大的pyltp我就体会到了,简直一个版本的pyltp需要一个版本的visio studio。我尝试了很多版本,最终选择了Tensorflow-gpu 1.14.0rc1 + cuda_8.0.61.2_windows + cudnn-8.0-windows10-x64-v6.0 首先下载cuda,之后默认路径安装后有如下文件就算安装成功了。
Tensorflow-GPU在window环境下安装历程_第1张图片
      然后下载cudnn,解压后有三个文件夹。
Tensorflow-GPU在window环境下安装历程_第2张图片
      将cudnn各个文件夹中的文件复制到对应CUDA的相同文件夹中。

      最后修改环境变量,安装CUDN后默认的环境变量是CUDA_PATH = C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0,没有显示bin目录和x64目录,单独将其添加到path中。最终还要确认有visio studio c++ 2015 Redistributable - x64环境。没有的话会报错 ‘DLL not found’.

     然后打开命令行输入pip install tensorflow-gpu,等待gpu版本的tensorflow安装完成就OK了,输入 python进入python环境,输入 import tensorflow as tf,没有报错就说明安装成功,如果报错,重启一次。

  • Tensorflow-GPU在window环境下安装历程_第3张图片
  • 大小: 39.4 KB
  • Tensorflow-GPU在window环境下安装历程_第4张图片
  • 大小: 47.8 KB
  • 查看图片附件

你可能感兴趣的:(python,tensorflow-gpu)