配置tensorflow-gpu，大过年的头秃啊

项目场景：

一年前，在老师电脑上配置成功过，准备在自己电脑上配置一下，处处碰壁啊。本人笔记本电脑配置如下：

Windows 10 Home，64bit

NVIDIA GeForce MX150

Visual Studio Community 2015

Python 3.6.5

keras 2.1.5

tensorflow 1.5.0

tensorflow-gpu 1.13.2

numpy

文章目录

---

配置步骤

一、安装Visual Studio Community 2015

一定要先装vs！！！

下载网址：https://visualstudio.microsoft.com/zh-hans/vs/older-downloads/

二、安装CUDA

1.查看电脑CUDA版本：控制面板—NVIDIA 控制面板—系统信息—组件

2. 下载相应版本CUDA Toolkit

下载网址：CUDA Toolkit Archive | NVIDIA Developer

3. 安装完成后，win+R —> cmd，输入命令nvcc -V ，若出现如下界面，则安装成功。

三、安装cuDNN

1.下载相应版本cuDNN

cuDNN Archive | NVIDIA Developer下载网址：cuDNN Archive | NVIDIA Developer

2. 下载后将文件解压，将以下文件替换到CUDA安装路径下

默认安装路径为C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0

3. 检查环境变量，确保有以下三个路径

路径：右击我的电脑—> 属性—>系统属性—>环境变量—>系统变量—>编辑环境变量

四、安装包

一定要对应版本！！！

1.这里用的是conda

创建虚拟环境

conda create -n tensorflow-gpu python=3.6

激活环境

activate tensorflow-gpu

安装包

conda install tensorflow-gpu==1.13.0

2. 使用pip安装

tensorflow安装命令

pip install tensorflow==1.5.0 -i https://pypi.tuna.tsinghua.edu.cn/simple/

-i https://pypi.tuna.tsinghua.edu.cn/simple/ 为临时清华园镜像

此外，也可以永久修改

在C盘User下创建一个pip.ini文件（路径：C:\Users\**\pip\），内容如下

pip install pip -U
pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple

 tensorflow-gpu安装命令

pip install tensorflow-gpu==1.13.2

keras安装命令

pip install kears==2.1.5

五、验证成功与否

import tensorflow as tf
import os

os.environ['TF_CPP_MIN_LOG_LEVEL']='2'
a = tf.constant(1.)
b = tf.constant(2.)
print(a+b)
print('GPU:', tf.test.is_gpu_available())

 

若出现以下界面，则配置成功！