Win10+GTX1060 安装配置 tensorflow


1 安装前须知

1.1 查看已安装显卡的运算能力

对于N卡来说,tensorflow-gpu的安装需要 CUDA 计算能力为 3.5 或更高的 NVIDIA 显卡,安装前最好先查看自己显卡的运算能力是否满足要求。进入https://developer.nvidia.com/cuda-gpus,根据自己显卡的产品系列进行查看。
对于 GTX1060,选择CUDA-Enabled GeForce and TITAN Products 按钮即可查看,嫌麻烦的可以直接看下图(只截取了部分显卡的运算能力图):
在这里插入图片描述
可以看到 GTX1060 的运算能力为 6.1,满足 tensorflow-gpu 的安装要求。

1.2 查看个软件的配套与版本要求

安装 tensorflow-gpu 还需要先安装Python、CUDA、cuDNN,但在安装上述依赖软件时还需根据软件的配套与版本要求进行下载安装。
在这里插入图片描述

1.3 没有 VS2015 或 VS2017 怎么安装tensorflow-gpu

很多博文提到了安装 CUDA 前需要安装 VS2017 或 VS2015,但这两个软件占用空间非常大,对于不做 C/C++ 开发的人,估计很多和我一样内心是拒绝的。好在通过单独下载安装上表中的 MSVC 2015 update3 也可以达到同样的目的。

1.4 显卡驱动需要安装吗

其实是不需要单独安装显卡驱动的,因为 CUDA 里自带了图像显示驱动,在安装 CUDA 时可以自动覆盖掉原来的显卡驱动。

1.5 依赖软件安装顺序

(1) Python ——> (2) MSVC 2015 update3 ——> (3) CUDA ——> (4) cuDNN ——> (5) tensorflow-gpu



2 下载软件

感谢别人的经验,我已安装成功。且其依赖软件的版本配套为:
Python 3.6.5 (Anaconda 5.2.0) + MSVC 2015 update3 (x64) + CUDA 9.0 + cuDNN 7.4.2 + tensorflow-gpu 1.8.0

下载地址如下:
Anaconda:https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/
MSVC 2015 update3:https://www.microsoft.com/en-us/download/details.aspx?id=53587
CUDA:https://developer.nvidia.com/cuda-toolkit-archive
cuDNN:https://developer.nvidia.com/rdp/cudnn-archive#a-collapse51b
tensorflow-gpu:pip install tensorflow-gpu==1.8.0(这一步直接会下载并安装,所以等上面四个软件安装完后再执行这一步)



3 安装软件

安装过程不是很顺利,踩了很多坑。感谢下面两篇比较好的博文把我从坑中拽了出来。
https://www.jianshu.com/p/62d414aa843e
https://blog.csdn.net/Canghaiyisu00/article/details/88292086



4 解除 FutureWarning 警告

装好 tensorflow-gpu 后,如果进行测试,可能会出现 FutureWarning 警告,可按照这篇博文提供的方法进行解决。



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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值