win10+anaconda+cuda+cudnn+TensorFlow-gpu

背景

笔者的笔记本显卡是1050。

原料

 Win10上搭建TensorFlow的开发环境需要至少需要安装3个软件,分别为:1.python,2.CUDA和CuDNN,3.TensorFlow(GPU版)。安装前一定要确认好每个软件的版本是否相互支持。

1. Python

相比于作为一个过渡版本的Python 2.6,笔者选择的是Python 3.6。在安装时,选择了Anaconda(一个开源的Python发行版本)的最新版本。Anaconda包含了conda、Python等180多个科学包及其依赖项,功能十分强大。
Anaconda下载地址: https://www.anaconda.com/download/

2. CUDA和CuDNN

CUDA是NVIDIA推出的运算平台,CuDNN是专门针对Deep Learning框架设计的一套GPU计算加速方案。在安装之前要查询下最新TensorFLow发行版支持到了哪个版本。笔者在安装TensorFLow时,CUDA已经到了9.1版本,但是TensorFLow1.7只支持到9.0版。另外,也要确认CUDA版本是否支持自己的显卡。笔者电脑的MX150只有CUDA9.0及以上的版本才支持。基于以上两个条件,笔者选择了CUDA9.0,并下载了对应的CuDNN版本。另外,笔者在后续的编程中发现tensorflow1.7只支持7.0的CuDNN。
1)显卡型号支持: https://developer.nvidia.com/cuda-gpus

2)CUDA下载地址:https://developer.nvidia.com/cuda-toolkit-archive


3)CuDNN下载地址:https://developer.nvidia.com/rdp/cudnn-download

(下载CuDNN需要注册账号,注册过程比较简单)



3. TensorFlow

TensorFlow的版本信息可以在Github,tensorflow中文社区以及pypi上查看。

    Github:https://github.com/tensorflow/tensorflow/releases

    tensorflow社区:https://tensorflow.google.cn/versions/

    pypi:https://pypi.org/project/tensorflow/#history

TensorFlow可以直接在Anaconda Prompt的命令行中用指令:“conda install tensorflow-gpu”直接安装,并且该指令在安装TensorFlow时还会顺带把Cuda和CuDNN也给装了。考虑到conda 的软件包并没有官方支持,并且tensorflow和附带的Cuda和CuDNN版本都不是最新的,笔者未使用conda安装,而是使用原生的 pip 安装,安装过程见后文。


安装过程

1. 安装Anaconda3 5.1

 Anaconda过程安装过程简单,打开安装包后选择好路径后就能安装


安装启动前若不勾选上第一项,则需在软件安装完成手动添加环境变量


2. 安装CUDA® Toolkit 9.0+cuDNN v7.1

1)CUDA9.0安装

运行cuda_9.0.176_win10.exe。安装软件会先运行一个系统检查,如果没有软硬件不兼容的情况就能继续进行下一步。若有不兼容情况,系统检查则会报错,无法进行下一步。如果报错了,就最先考虑下是不是自己的显卡不被该版本支持。


在选项着一栏选择自定义,否则安装程序会一股脑把里面所有软件包都给你装上。同时安装路径有三个,自己新建一个文件夹cuda,子文件夹cuda-v9.0;子文件夹cudasample-v9.0,然后选路径时cudasample选到cudasample-v9.0,另外两个选到cuda-v9.0。


在选择安装项时一般不安装GeForce Experience,CUDA是核心组件必须勾上,其他的先更新一下显卡驱动就可以了,不用全装。只用选第一个cuda就可以了。


2)CuDNN7.1安装

解压压缩包cudnn-9.0-windows10-x64-v7.0.zip,得到三个文件夹


将这三个文件夹中的文件拷贝到CUDA9.0的安装路径的对应根文件夹下



3. 安装tensorflow1.7

打开Anaconda navigator,进入环境配置页面,直接搜索TensorFlow-gpu,点击安装即可。


验证

在命令行中,进入python,并输入以下代码:
import tensorflow as tf
hello = tf.constant(‘Hello, TensorFlow!’)
sess = tf.Session()
print(sess.run(hello))
输出:
b’ Hello, TensorFlow!

            </div>
  • 0
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
### 回答1: 要在Windows 10上安装tensorflow-gpu,可以按照以下步骤进行操作: 1. 安装CUDA Toolkit TensorFlow-gpu需要CUDA Toolkit作为其GPU支持的基础。因此,首先需要在你的计算机上安装适用于你的GPU版本的CUDA Toolkit。可以从NVIDIA官网下载相应的版本,并按照安装向导进行操作。 2. 安装cuDNN TensorFlow-gpu还需要cuDNN作为其GPU支持的一部分。cuDNN是NVIDIA的深度学习库,可以通过NVIDIA的网站下载并安装。 3. 创建虚拟环境 推荐在安装tensorflow-gpu之前创建一个Python虚拟环境。可以使用Anaconda或者Python自带的venv模块来创建虚拟环境。 4. 安装tensorflow-gpu 在虚拟环境中,可以使用pip来安装tensorflow-gpu,命令为:pip install tensorflow-gpu。 完成以上步骤后,就可以在Windows 10上使用tensorflow-gpu了。 ### 回答2: win10安装tensorflow-gpu TensorFlow是一个开源的机器学习库,它由谷歌公司开发和维护。它具有高效、灵活、易用等优点,是构建人工智能应用的主要工具之一。如果你使用GPU进行深度学习,那么安装tensorflow-gpu可以提高训练速度,让你更快地处理大规模的数据集。在本文中,我们将介绍如何在Windows 10操作系统上安装tensorflow-gpu。 1. 安装CUDA工具包 如果你正在使用NVIDIA GPU,那么你需要先安装CUDA工具包。CUDA是一款NVIDIA推出的并行计算平台和编程模型,它支持使用GPU进行高性能计算。在安装CUDA之前,你需要确定你的GPU是否支持CUDA。可以在NVIDIA官网上查看你的GPU支持的CUDA版本。接下来,按照如下步骤安装CUDA 打开NVIDIA官网,选择CUDA下载页面 选择符合你GPU的版本,下载相关的安装包 安装CUDA,按照提示进行操作 2. 安装cuDNNcuDNN库是一个加速深度神经网络的库,它也是由NVIDIA进行开发和维护。为了确保你的深度学习应用能够最大限度地利用GPU加速,你需要安装cuDNN库。 到NVIDIA官网上下载cuDNN库,下载之前需要先注册一个账号 下载完成后,解压文件,将文件的路径添加到PATH环境变量中 3. 安装Anaconda Anaconda是一个Python发行版,它包括了Python解释器、常用的Python包和工具。安装Anaconda可以使我们更加方便地安装、管理Python包和环境。在安装Anaconda之前,你需要选择你想要的Python版本。你可以在Anaconda官网上下载适合你的版本。 安装Anaconda,将其添加到系统环境变量中。安装过程中需要注意安装路径,如果路径中带有空格,请把路径加到环境变量时加上引号。 4. 安装TensorFlow 打开Anaconda Prompt(或使用其他命令行工具),输入如下命令: conda create -n tensorflow-gpu python=3.6 activate tensorflow-gpu pip install tensorflow-gpu 这个命令将会在你的Anaconda中创建一个名为tensorflow-gpu的新环境,安装Python版本为3.6,然后安装包含TensorFlow-gpu的依赖。 安装完成后,你需要运行一些TensorFlow的示例代码以确保安装成功。打开Python和TensorFlow的交互式环境,输入如下命令: import tensorflow as tf hello = tf.constant('Hello, TensorFlow!') sess = tf.Session() print(sess.run(hello)) 如果输出结果是“Hello, TensorFlow!”,那么恭喜你,你已经成功安装了TensorFlow-gpu。 总结来说,win10安装tensorflow-gpu的步骤主要包括安装CUDA工具包、安装cuDNN库、安装Anaconda以及安装TensorFlow。毫无疑问,这是一项需要花费一些时间和精力的任务,但是一旦你成功地完成了上述步骤,你将可以轻松地使用GPU进行深度学习,以及更快地处理大规模的数据集。 ### 回答3: Windows 10作为一个非常流行的操作系统,在计算机领域中被广泛使用。如果你想要在Windows 10上使用tensorflow-gpu,那么你需要做一些准备工作,以确保安装能够成功进行。下面是一些安装步骤: 1. 下载Anaconda 在安装tensorflow-gpu之前,需要先下载AnacondaAnaconda是一个开源的Python发行版,这里我们需要下载Python 3.6版本的Anaconda。下载的链接为:https://www.anaconda.com/products/individual#Downloads,选择适合自己的版本进行下载并安装。 2. 安装CUDA 在下载tensorflow-gpu之前,需要先安装CUDACUDA是一个并行计算平台,它可以加速机器学习和深度学习的计算过程。在这里,我们需要安装的是CUDA 9.0版本。首先,需要前往官网下载CUDA 9.0版本的安装包,链接为:https://developer.nvidia.com/cuda-90-download-archive。下载并按照默认设置进行安装。 3. 安装cuDNN cuDNN是用于加速深度学习计算的一个库。但是,在安装之前,需要先注册一个NVIDIA账号。注册完成之后,前往链接https://developer.nvidia.com/cudnn下载cuDNN的安装包。根据下载安装文件的说明安装即可。 4. 安装tensorflow-gpu 安装tensorflow-gpu之前,需要先打开Anaconda Prompt,并创建一个新的虚拟环境,输入以下命令: conda create --name tensorflow-gpu python=3.6 这个命令将会创建一个名为tensorflow-gpu的虚拟环境,并使用Python 3.6版本。 安装完成之后,可以激活这个虚拟环境,输入以下命令: conda activate tensorflow-gpu 接下来就可以安装tensorflow-gpu了,输入以下命令: pip install tensorflow-gpu 安装完成后,我们可以通过以下语句验证tensorflow-gpu是否正常安装: python -c "import tensorflow as tf; tf.enable_eager_execution(); print(tf.reduce_sum(tf.random_normal([1000, 1000])))" 如果安装成功,将会有一大串数字输出,表示tensorflow-gpu已经成功安装。 综上,如果你想在Windows 10上安装tensorflow-gpu,需要先下载Anaconda,然后安装CUDAcuDNN,最后安装tensorflow-gpu。请按照以上步骤进行操作。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值