win10 + cuda(v9.0) 安装TensorFlow-gpu版

2024-08-25 10:46:12

之前在实习公司的电脑上装过TensorFlow-gpu，那时候很快就装好了。但在自己的笔记本上装时，却搞了很久。。。

一部分原因是因为用校园网下载cuda toolkit 和cudnn ，总是在最后时刻罢工。辗转叫了几个人帮忙，才把这两个东西下载好。。

安装步骤

其实，这个安装过程还是蛮简单的。据说cuda版本和cudnn版本没选好的话，就会遇到很多坑，所幸，我选的版本都能很好的驾驭tensorflow。

下面是要安装的东西，可以依次安装，也可以按你喜欢的顺序来安装。但要保证安装了VC++插件和python之后，才安装tensorflow。

Visual C++ Redistributable for Visual Studio 2015： tensorflow其实是基于VC++2015开发的，所以需要安装Visual C++ Redistributable for Visual Studio 2015。下载后，直接默认安装就行
python环境：我用的是anaconda3.x，新建了一个python3.5的环境，然后在python3.5中安装的tensorflow-gpu。这个不会的话，可以尝试搜索“anaconda安装、anaconda清华源安装”和“anaconda新建python环境”。。
cuda9.0： cuda-v9.0.下载之后，默认安装就行，中间要选同意的地方就选同意。

安装好cuda后，要配置系统环境。用小娜搜索“控制面板”–>选择大图标–>选择系统–>选择高级系统设置–>选择环境变量–>选择path，新增加“C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0”–>确定。。然后再新增几个变量，如下图（已经存在的就不用增加了）：
cudnn：（cuDNN v7.1.4 Library for Windows 10）下载好后，解压，把所有东西复制到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0 中，就ok啦
tensorflow-gpu：安装win+R，运行cmd，打开命令窗口，激活python3.5的环境（activate 环境名），然后输入pip install tensorflow-gpu 即可。等待安装成功就行。

测试

我喜欢用jupyter notebook来写python，所以现在也用它来写一下测试代码：

import tensorflow as tf

hi = tf.constant('Hi , welcome to TensorFlow ! ')

sess = tf.Session()

print(sess.run(hi).decode())

得到结果如下：

cool，顺利输出了结果，说明安装成功。

最新文章

热门文章