首页 技术 正文
技术 2022年11月17日
0 收藏 491 点赞 3,846 浏览 1823 个字

一、检查显卡

·查看自己的显卡配置是否能支持cuda,以及Tensorflow不同版本要求与CUDA及CUDNN版本对应关系:

https://developer.nvidia.com/zh-cn/cuda-gpus

·nvidia与cuda需要满足关系:

https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html

二、下载安装GeForce Experience

官网下载:

https://www.nvidia.com/zh-cn/geforce/geforce-experience/,

experience检测显卡驱动,发现最新版本472.12,点击安装,一定要注意,选择自定义安装,然后有个纯净安装的勾选框。接下来会让重启电脑,重启过后experience还会继续弹出来自动安装一些东西,等它安装结束即可。

三、下载 CUDA 8.0 + cudnn 5.1

· cuda历史版本下载连接:

版本是:CUDA Toolkit 8.0 GA2 (Feb 2017)

https://developer.nvidia.com/cuda-toolkit-archive

· cuda与cudnn需要满足关系:

版本是:cuDNN v5.1 Library for Windows 10

https://developer.nvidia.com/rdp/cudnn-archive

· 显卡计算能力:

https://developer.nvidia.com/cuda-gpus

对应版本:

由上表可知,cuda8+cudnn5.1最高可以装到tensorflow1.2.0)

四、安装

1. CUDA 8.0

直接点击安装

在cmd命令行中输入nvcc -V查看是否安装成功。

2. 安装cudnn 5.1

cudnn下载完成后直接解压,然后到你cuda的安装路径下,

我的cuda路径为C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA

· 将解压出来的:

bin中文件放到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\bin中

include文件放到cuda的C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\include中

lib文件放到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\lib\x64

3. 安装 Tensorflow_gpu 1.0.0

· 首先打开Anaconda Prompt,进入Anaconda命令行管理界面。

· 配置清华镜像,输入指令:

conda config –add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/

· 然后输入

conda config –set show_channel_urls yes

· 创建运行环境,输入指令:

conda create -n tensorflow-gpu python=3.6

· 新建一个名字叫“tensorflow-gpu”,python版本为3.6的运行环境,此环境与Anaconda中其它环境隔离。红框中的软件包也会随之安装,· · 输入“y“和回车后开始安装。

· 然后会出现一些软件安装包的提示(问你是否确认安装):输入 y 按回车

· 激活并进入环境,使后续指令在激活的环境中生效,输入指令:

activate tensorflow-gpu

· 升级pip到最新版,防止稍后的安装时,出现错误(很有可能因为没有更新pip到最新版,导致下载到一半出现错误),输入指令:

python -m pip install –upgrade pip

· 安装tensorflow 1.2.0及相应依赖包,输入指令:

pip install –ignore-installed –upgrade tensorflow-gpu

至此,tensorflow的安装完成。

笔者安装过程截图:

· 最后,验证一下是否安装成功:

在命令行中,进入python,并输入以下代码:

import tensorflow as tf

hello = tf.constant(‘Hello, TensorFlow!’)

sess = tf.Session()

print(sess.run(hello))

输出:

b’ Hello, TensorFlow!

相关推荐
python开发_常用的python模块及安装方法
adodb:我们领导推荐的数据库连接组件bsddb3:BerkeleyDB的连接组件Cheetah-1.0:我比较喜欢这个版本的cheeta…
日期:2022-11-24 点赞:878 阅读:9,088
Educational Codeforces Round 11 C. Hard Process 二分
C. Hard Process题目连接:http://www.codeforces.com/contest/660/problem/CDes…
日期:2022-11-24 点赞:807 阅读:5,564
下载Ubuntn 17.04 内核源代码
zengkefu@server1:/usr/src$ uname -aLinux server1 4.10.0-19-generic #21…
日期:2022-11-24 点赞:569 阅读:6,412
可用Active Desktop Calendar V7.86 注册码序列号
可用Active Desktop Calendar V7.86 注册码序列号Name: www.greendown.cn Code: &nb…
日期:2022-11-24 点赞:733 阅读:6,185
Android调用系统相机、自定义相机、处理大图片
Android调用系统相机和自定义相机实例本博文主要是介绍了android上使用相机进行拍照并显示的两种方式,并且由于涉及到要把拍到的照片显…
日期:2022-11-24 点赞:512 阅读:7,822
Struts的使用
一、Struts2的获取  Struts的官方网站为:http://struts.apache.org/  下载完Struts2的jar包,…
日期:2022-11-24 点赞:671 阅读:4,905