Tensorflow-gpu环境搭建

显卡设备:英伟达1060

系统:Windows10

python版本:3.7.4

CUDA:cuda_10.0.130_411.31_win10

cudnn:cudnn-10.0-windows10-x64-v7.6.5.32

tensorflow:gpu1.14

一 安装python3.7.4

  https://www.python.org/downloads/windows/ 下载安装

二 安装cuda:

  1.首先更新显卡驱动;

  2.桌面右键-NVIDIA控制面板-帮助-系统信息-组件,可以查看当前设备支持的最高cuda版本。

  我更新驱动之前支持最高到8,更新后就支持10版本了。

  

 

  3.下载cuda

  https://developer.nvidia.com/cuda-downloads 

  安装。一直下一步就行。选择组件的时候可以不选择显卡驱动,如果你的驱动比较新。

  4.环境变量

  

 

 三 cudnn安装

  1 下载https://developer.nvidia.com/rdp/cudnn-archive cuda和cudnn有对应版本关系。

  2.解压文件夹,将解压后的文件夹下的文件拷贝到cuda安装目录下,与之相对应的文件夹下。

    解压后其实是三个文件夹,合并到cuda的安装目录中就可以了。

    默认路径是C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0 找不到了可以类比找一下

四 安装tensorflow-gpu

  cmd中直接运行

  pip install -i https://pypi.tuna.tsinghua.edu.cn/simple tensorflow-gpu==1.14

    

 

 

posted @ 2019-12-07 12:05  Milburn  阅读(1143)  评论(0编辑  收藏  举报