深度学习Tensorflow主机环境配置: Win10+Nvidia GTX 1080i+CUDA8.0+CUDDN6

本文详细介绍了如何在Windows 10系统上配置Tensorflow GPU环境,包括硬件配置、系统要求和软件安装步骤,特别是针对CUDA 8.0和cuDNN v6.0的安装及验证过程。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

环境说明

硬件:

CPU:E5-2673 V3

主板:华硕Z10Z10PA-D8

内存:三星DDR4 2400 ECC-R 16G*2

显卡:微星GTX1080Ti AERO

固态硬盘:三星PM961 256G M.2

机械硬盘:希捷2T

电源:长城1250W

机箱:ANTEC P8

风扇:10热管全静音

系统:Windows 10 Enterprise Version

前期工作

NVIDIA 最新版驱动

Visual Studio 2015 (需要C++部分,可以安装visual studio community 2015

备注:经过测试,Visual Studio 2013也可以。

安装CUDA和cuDNN

CUDA的版本要和自己的显卡要对应,CUDA是显卡驱动和cundd(用来加速深度学习训练库)版本要对应,在训练网络时,使用GPU加速会调用CUDA驱动和cudnn库

注意:发这篇文章的时候,tensorflow尚不支持最新版本的CUDA和cuDNN,因此,本文使用以下两个版本:

1. CUDA 8.0

下载地址:developer.nvidia.com/cu

直接安装即可。也可以到我的百度云下载:链接:pan.baidu.com/s/1dFHD37 密码:u0gi

CUDA系统会自动安装到C盘的Program Files\NVIDIA GPU Computing Toolkit文件中,

2.cuDNN v6.0

下载地址:developer.nvidia.com/rd

也可以到我的百度云下载:链接:pan.baidu.com/s/1gftmqR 密码:zinu

解压后覆盖到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0目录即可。,这时要特别注意用cudnn三个文件中的文件,复制到CUDA相应的文件夹中(仔细看一下他们的目录结构完全一致)

3. 安装Tensorflow GPU 1.4

由于Anaconda 可以提供完整的科学计算库,所以直接使用Anaconda 来进行相关的额安装。

具体命令为:pip install tensorflow-gpu

如果要安装最新版本,可输入:pip install -upgrade tensorflow-gpu,也可以在第5步再安装

4.安装Anaconda

下载地址:anaconda.com/download/

这里我们下载Python 3.6 64bit 的Anaconda 5,直接安装即可。

5. 在Anaconda中安装TensorFlow GPU 1.4

conda create –name tf python=3.6 #创建tf环境
activate tf #激活tf环境

pip install –ignore-installed –upgrade tensorflow-gpu

其他命令:

deactivate tf #退出tf环境
conda remove –name tf –all #删除tf环境(全部删除)

将会自动安装如下组件

numpy
wheel
tensorflow-tensorboard
six
protobuf
html5lib
markdown
werkzeug
bleach
setuptools

使用下列代码测试安装正确性

6.测试.

命令行下:

activate tf

python

输入以下代码:

import tensorflow as tf
hello = tf.constant(‘Hello, TensorFlow!’)
sess = tf.Session()
print(sess.run(hello))

没有报错就是配置好了。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值