基于深度学习的建筑能耗预测02——安装Tensorflow-gpu

数字设计-基于深度学习的建筑能耗预测—2021WS
作者:徐仔
指导老师:万老师、丁老师
带组导师:宋学姐(建筑学)、卢学长(计算机)
(转载请注明出处)

一、检查显卡

·查看自己的显卡配置是否能支持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.2.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'
即安装成功!

posted @ 2021-09-29 22:59  徐仔小天才  阅读(185)  评论(0编辑  收藏  举报