window下GPU版tensorflow安装

6 篇文章 0 订阅
4 篇文章 0 订阅

<1>.首先下载好三个软件备用:

1. Anaconda

2. cuda

3. cuDNN压缩包

如下图所示:
这里写图片描述

<2>.然后点击Anaconda进行安装

中途要选择好安装目录,一直点下一步,于如下步骤作如下选择:

这里写图片描述
继续下一步直至结束。

<3>.配置Anaconda环境变量,如下方式:

这里写图片描述

path变量的值具体如下所示,需要根据你的anaconda具体的安装目录设定:
 1. D:\Albert\simul\InstallPlace;
 2. D:\Albert\simul\InstallPlace\Scripts;
 3. D:\Albert\simul\InstallPlace\Library\bin

<4>.点击cuda进行安装,一直为下一步即可

中途遇到选择安装程序时,可以只选择安装computing toolkit程序。
中途遇到选择安装路径时,可以自行设定。

<5>.点击解压cuDNN

将其中bin,include,lib中内容复制到D:\Albert\simul\CUDA\CUDA_Development(这是我的安装路径)中对应目录中。

<6>.在命令行窗口中输入:

pip3 install --upgrade tensorflow-gpu
完成。

*后记

如果此时输入

import tensorflow as tf
tf.Session()

可能会出现信息:

cudaGetDevice() failed:CUDA driver version is insufficient for CUDA runtime version
failed to create session

这是说明CUDA版本与显卡驱动版本不匹配,此时可以登录网站https://www.nvidia.cn/Download/index.aspx?lang=cn获取显卡最新驱动的version。

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值