首页 > 其他 > 详细

tensorflow-gpu安装

时间:2019-12-07 14:43:36      阅读:96      评论:0      收藏:0      [点我收藏+]

显卡设备:英伟达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

    

 

 

tensorflow-gpu安装

原文:https://www.cnblogs.com/Milburn/p/12001264.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!