tensorflow-gpu安装(win10)

下载对应版本的cuda,cudNN,anaconda(python3.7)
1、下载cuda_10.0.130_411.31_win10.exe,安装过程默认安装即可,需要注意的是,有时如果你的电脑里有VS,但CUDA提示你需要再次安装,可以在cuda/选项/自定义安装中取消勾选VS,其余过程很简单
2、下载对应版本的cudNN,在cudnn官网上有对应cuda的版本(需要注册账号)
解压之后,得到四个文件,分别为bin,include,lib,**txt
将三个文件夹包含的文件(注意:不是文件夹)放到cuda对应文件夹下
即将:bin中的cudnn64_7.dll     放到      NVIDIA GPUComputing Toolkit/CUDA/v10.0/bin
           include中的cudnn.h        放到      NVIDIA GPUComputing Toolkit/CUDA/v10.0/include
           lib/x64中的cudnn.lib       放到      NVIDIA GPUComputing Toolkit/CUDA/v10.0/lib/x64
           **.txt                                放到      NVIDIA GPUComputing Toolkit/CUDA/v10.0中即可
    如果版本不对应,在安装tensorflow-gpu会提示你找不到指定文件,例如我先下载了一个10.1版本的cuda,在安装时显示,未找到指定的NVIDIA GPUComputing Toolkit/CUDA/v10.0/bin/cudart64_100.dll。因为我的10.1版本中只有cudart64_101.dll。总之一定要版本对应。
3、anaconda prompt中输入pip install -i https://pypi.tuna.tsinghua.edu.cn/simple tensorflow-gpu==1.14.0
等待一会,然后就可以安装完成。

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值