Tensorflow-gpu环境搭建详细教程(内含cuda和cudnn安装教程)
Tensorflow-gpu环境搭建详细教程
配置环境
安装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
2.选择你的windows系统版本,我的是window10
3.选择第一个选项
4.选择第一个开始下载
5.下载完之后打开安装
6.选择自定义选项
7.选择CUDA
8.下一步
9.点击结束,安装完成
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中去(这里是默认的安装路径)
下载安装CUDNN9.0
cudnn9.0 官方下载地址.
当你进入官网下载时,需要你注册登录后才能下载,我们按照步骤注册登录就好了
1.点击Download cuDNN
2.点击login
3.点击CREATE AN ACCOUNT注册
4.注册完后登录,完成一个问卷后点击I agree和Archived cuDNN Releases 查看更多版本
5.选择cudnn9.0版本
6.选择对应window10的版本
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.得到下图则安装完成
下载安装anaconda3
anaconda 各版本下载地址.
1.选择anaconda3-5.3.1-windows-86_64版本开始下载
2.下载后打开进行安装
3.选择第一个
4.自己选择安装路径
5.两个选项都选上,点击Install下载
6.打开Anaconda Navigator
7.检查Anaconda是否成功安装:conda --version
8.检测目前安装了哪些环境:conda info --envs
9.检查目前有哪些版本的python可以安装:conda search --full-name python
10.用anaconda创建一个虚拟环境:conda create --name tensorflow python=3.6(中间根据提示输入y)
11.按照提示,激活环境:conda activate tensorflow-gpu
12.确保名叫tensorflow-gpu的环境已经被成功添加:conda info --envs
Tensorflow的安装
1.用豆瓣镜像去安装tensorflow-gpu1.9.0:
pip install --index-url https://pypi.douban.com/simple tensorflow-gpu==1.9.0
测试
1.打开Anaconda Prompt,进入虚拟环境:conda activate tensorflow-gpu
2.输入python,回车
3.然后输入import tensorflow as tf
hello = tf.constant(‘hello tensorflow’)
sess = tf.Session()
print(sess.run(hello))
结果:b’Hello tensorfolw’,则安装成功。
遇到的问题
1.pip版本过低
根据提示,使用python -m pip install --upgrade pip升级失败
使用pip install --user --upgrade pip成功升级
确认pip是否升级成功:pip show pip
2.如果出现镜像下载超时的问题,换个镜像下载即可。
本文地址:https://blog.csdn.net/uran_orange/article/details/107644151