Windows10下安装TensorFlow-GPU(1.1.0)

1. 安装前确认工作

  • 确保电脑显卡是Nvidia显卡,并且支持CUDA;
  • 系统中存在一个64位的版本为3.5(只能为3.5或者3.4)的Python;
  • 确保系统拥有Visual C++ Redistributable (x64) 2015以上版本;
  • 已经拥有Anaconda环境(需要它是因为需要创建一个独立出来的环境,非必须).

2.安装环境

由于电脑中的Anaconda的Python版本是3.6的,所以这里就利用conda构建了一个3.5的环境。

1、用管理员身份打开cmd窗口,输入命令conda create --name python35 python=3.5,等待安装完成即可;

2、此时已经在Anaconda安装目录的envs文件夹下创建了一个独立与系统原有的版本的Python,在cmd下输入命令activate python35即可激活它(退出环境使用deactivate python35命令);

3、进入新环境之后,键入命令pip install tensorflow-gpu安装TensorFlow,等待安装完成;

Windows10下安装TensorFlow-GPU(1.1.0)_第1张图片

4、虽然已经安装好了TensorFlow,但是需要使用GPU进行加速,所以需要安装NVIDIA相应的CUDA和cuDnn环境;
5、需要的CUDA版本是v8.0的,cuDnn版本是v5.1的,先前往下列网站下载CUDA v8.0,https://developer.nvidia.com/cuda-downloads ,进行如下图的选择下载即可。下载完成后,完成安装即可;

Windows10下安装TensorFlow-GPU(1.1.0)_第2张图片
Windows10下安装TensorFlow-GPU(1.1.0)_第3张图片
Windows10下安装TensorFlow-GPU(1.1.0)_第4张图片
Windows10下安装TensorFlow-GPU(1.1.0)_第5张图片
Windows10下安装TensorFlow-GPU(1.1.0)_第6张图片

6、同时,进入一下网站https://developer.nvidia.com/rdp/cudnn-download 下载cuDnn v5.1。值得注意的是,下载cuDnn之前,需要进行NVIDIA网站帐号的注册和相关问卷的填写,完成之后即可转到下载网页;

Windows10下安装TensorFlow-GPU(1.1.0)_第7张图片

7、解压下载好的cudnn-8.0-windows10-x64-v5.1.zip文件,将cuda下的三个文件夹复制到CUDA的安装目录下,如图所示覆盖即可;

Windows10下安装TensorFlow-GPU(1.1.0)_第8张图片
Windows10下安装TensorFlow-GPU(1.1.0)_第9张图片

8、此时打开系统环境变量,设置CUDA_PATH,如下图所示即可;

Windows10下安装TensorFlow-GPU(1.1.0)_第10张图片
Windows10下安装TensorFlow-GPU(1.1.0)_第11张图片

9、由此,完成安装与设置工作。

3.测试

1、进入cmd激活python35环境,键入python进入python shell;

2、输入import tensorflow as tf导入tensorflow库,无报错即成功安装TensorFlow;

3、键入代码

hello = tf.constant('Hello, TensorFlow!')
sess = tf.Session()

有如下图的效果,即说明使用了GPU进行计算。


Windows10下安装TensorFlow-GPU(1.1.0)_第12张图片

你可能感兴趣的:(Windows10下安装TensorFlow-GPU(1.1.0))