超详细:win10安装tensorflow-gpu1.8.0完整步骤

本文详述了在Windows 10系统中安装TensorFlow-GPU 1.8.0、CUDA 9.0和cuDNN v7.1的完整步骤,包括检查GPU兼容性、Anaconda安装、避免Visual Studio Integration错误以及配置环境变量等关键环节。
摘要由CSDN通过智能技术生成

在整个安装的过程中也遇到了很多的坑,故此做个记录,争取下次不再犯!

我的整个基本配置如下:

电脑环境如下:win10(64位)+CPU:E5-2603 +GPU:GTX 1070
需要安装的软件如下:Anaconda3-4.2.0-Windows-x86_64(python 3.5.2) + tensorflow-gpu 1.8 + CUDA 9.0 + cuDNN v7.1 for CUDA9.0

若你想在自己的windows上安装tensorflow-gpu,一般化也可以遵循如下的步骤。

1.首先确定自己电脑的gpu是否支持cuda

能否支持tensorflow-gpu版本,这里主要是Nvidia显卡,如下网址:
https://developer.nvidia.com/cuda-gpus 基本上只要显卡在 GeForce 410M 以上都可以。

2.在显卡支持的情况下,主要是看tensorlfow-gpu和cuda以及cuDNN的对应关系

首先,cuda与cuDNN是相互对应的,所以,现在主要是看tensorflow与cuda的对应关系。因为从Tensorflow-gpu 1.6开始使用CUDA9.0+cuDNN7组合,不过好像不支持CUDA 9.1与CUDA 9.2。所以,最终我的选择是 tensorflow-gpu 1.8 + CUDA 9.0 + cuDNN v7.1

评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值