欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  IT编程

Tensorflow-gpu环境搭建详细教程(内含cuda和cudnn安装教程)

程序员文章站 2022-03-26 23:14:17
Tensorflow-gpu环境搭建详细教程配置环境下载安装CUDACUDA9.0的环境变量配置下载安装CUDNN9.0下载安装anaconda3Tensorflow的安装测试遇到的问题配置环境安装tensorflow-gpu之前,我们需要准备好环境anaconda3+cuda9.0+cudnn9.0+tensorflow-gpu1.9.0.其中cuna9.0和cudnn9.0和tensorflow-gpu1.9.0三者的版本需要相互对应。下载安装CUDAcuda9.0 官方下载地址.1.我这里...

配置环境

安装tensorflow-gpu之前,我们需要准备好环境anaconda3+cuda9.0+cudnn9.0+tensorflow-gpu1.9.0.
其中cuna9.0和cudnn9.0和tensorflow-gpu1.9.0三者的版本需要相互对应。

下载安装CUDA

cuda9.0 官方下载地址.
1.我这里用的是windows系统,选择window
Tensorflow-gpu环境搭建详细教程(内含cuda和cudnn安装教程)
2.选择你的windows系统版本,我的是window10
Tensorflow-gpu环境搭建详细教程(内含cuda和cudnn安装教程)
3.选择第一个选项
Tensorflow-gpu环境搭建详细教程(内含cuda和cudnn安装教程)
4.选择第一个开始下载
Tensorflow-gpu环境搭建详细教程(内含cuda和cudnn安装教程)
5.下载完之后打开安装
Tensorflow-gpu环境搭建详细教程(内含cuda和cudnn安装教程)
Tensorflow-gpu环境搭建详细教程(内含cuda和cudnn安装教程)
6.选择自定义选项
Tensorflow-gpu环境搭建详细教程(内含cuda和cudnn安装教程)
7.选择CUDA
Tensorflow-gpu环境搭建详细教程(内含cuda和cudnn安装教程)
8.下一步
Tensorflow-gpu环境搭建详细教程(内含cuda和cudnn安装教程)
9.点击结束,安装完成
Tensorflow-gpu环境搭建详细教程(内含cuda和cudnn安装教程)

CUDA9.0的环境变量配置

此电脑上点右键,打开属性->高级系统设置->环境变量,可以看到系统中多了CUDA_PATH和CUDA_PATH_V9_0两个环境变量
接下来需要将
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\lib\x64
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\bin
两个路径复制到环境变量path中去(这里是默认的安装路径)
Tensorflow-gpu环境搭建详细教程(内含cuda和cudnn安装教程)
Tensorflow-gpu环境搭建详细教程(内含cuda和cudnn安装教程)

下载安装CUDNN9.0

cudnn9.0 官方下载地址.
当你进入官网下载时,需要你注册登录后才能下载,我们按照步骤注册登录就好了
1.点击Download cuDNN
Tensorflow-gpu环境搭建详细教程(内含cuda和cudnn安装教程)
2.点击login
Tensorflow-gpu环境搭建详细教程(内含cuda和cudnn安装教程)
3.点击CREATE AN ACCOUNT注册
Tensorflow-gpu环境搭建详细教程(内含cuda和cudnn安装教程)
4.注册完后登录,完成一个问卷后点击I agree和Archived cuDNN Releases 查看更多版本
Tensorflow-gpu环境搭建详细教程(内含cuda和cudnn安装教程)
5.选择cudnn9.0版本
Tensorflow-gpu环境搭建详细教程(内含cuda和cudnn安装教程)
6.选择对应window10的版本
Tensorflow-gpu环境搭建详细教程(内含cuda和cudnn安装教程)
10.下载后解压,分别将.h .lib 和.dll 文件分别拷贝到cuda的include, lib/x64, bin 文件夹下,也就是对应的文件夹。

11.配置完成后,我们可以验证是否配置成功,主要使用CUDA内置的deviceQuery.exe 和 bandwithTest.exe
win+R启动cmd,cd到安装目录下的 …\extras\demo_suite,然后分别执行bandwidthTest.exe和deviceQuery.exe.得到下图则安装完成
Tensorflow-gpu环境搭建详细教程(内含cuda和cudnn安装教程)
Tensorflow-gpu环境搭建详细教程(内含cuda和cudnn安装教程)

下载安装anaconda3

anaconda 各版本下载地址.
1.选择anaconda3-5.3.1-windows-86_64版本开始下载
Tensorflow-gpu环境搭建详细教程(内含cuda和cudnn安装教程)
2.下载后打开进行安装
Tensorflow-gpu环境搭建详细教程(内含cuda和cudnn安装教程)
3.选择第一个
Tensorflow-gpu环境搭建详细教程(内含cuda和cudnn安装教程)
4.自己选择安装路径
Tensorflow-gpu环境搭建详细教程(内含cuda和cudnn安装教程)
5.两个选项都选上,点击Install下载
Tensorflow-gpu环境搭建详细教程(内含cuda和cudnn安装教程)
6.打开Anaconda Navigator
Tensorflow-gpu环境搭建详细教程(内含cuda和cudnn安装教程)
7.检查Anaconda是否成功安装:conda --version
Tensorflow-gpu环境搭建详细教程(内含cuda和cudnn安装教程)
8.检测目前安装了哪些环境:conda info --envs
Tensorflow-gpu环境搭建详细教程(内含cuda和cudnn安装教程)
9.检查目前有哪些版本的python可以安装:conda search --full-name python
Tensorflow-gpu环境搭建详细教程(内含cuda和cudnn安装教程)
10.用anaconda创建一个虚拟环境:conda create --name tensorflow python=3.6(中间根据提示输入y)
Tensorflow-gpu环境搭建详细教程(内含cuda和cudnn安装教程)
11.按照提示,激活环境:conda activate tensorflow-gpu
Tensorflow-gpu环境搭建详细教程(内含cuda和cudnn安装教程)
12.确保名叫tensorflow-gpu的环境已经被成功添加:conda info --envs
Tensorflow-gpu环境搭建详细教程(内含cuda和cudnn安装教程)

Tensorflow的安装

1.用豆瓣镜像去安装tensorflow-gpu1.9.0:
pip install --index-url https://pypi.douban.com/simple tensorflow-gpu==1.9.0
Tensorflow-gpu环境搭建详细教程(内含cuda和cudnn安装教程)

测试

1.打开Anaconda Prompt,进入虚拟环境:conda activate tensorflow-gpu
Tensorflow-gpu环境搭建详细教程(内含cuda和cudnn安装教程)
2.输入python,回车
Tensorflow-gpu环境搭建详细教程(内含cuda和cudnn安装教程)

3.然后输入import tensorflow as tf
Tensorflow-gpu环境搭建详细教程(内含cuda和cudnn安装教程)
hello = tf.constant(‘hello tensorflow’)
sess = tf.Session()
Tensorflow-gpu环境搭建详细教程(内含cuda和cudnn安装教程)
print(sess.run(hello))
Tensorflow-gpu环境搭建详细教程(内含cuda和cudnn安装教程)
结果:b’Hello tensorfolw’,则安装成功。

遇到的问题

1.pip版本过低
Tensorflow-gpu环境搭建详细教程(内含cuda和cudnn安装教程)
根据提示,使用python -m pip install --upgrade pip升级失败
Tensorflow-gpu环境搭建详细教程(内含cuda和cudnn安装教程)
使用pip install --user --upgrade pip成功升级
Tensorflow-gpu环境搭建详细教程(内含cuda和cudnn安装教程)
确认pip是否升级成功:pip show pip
Tensorflow-gpu环境搭建详细教程(内含cuda和cudnn安装教程)
2.如果出现镜像下载超时的问题,换个镜像下载即可。

本文地址:https://blog.csdn.net/uran_orange/article/details/107644151