我安装的是:WINDOWS10+CUDA10.0+CUDNN7.4+tensorflow-gpu1.14.0
1、首先选好自己要装的版本
在这里可以看到不同版本的tensorflow-gpu对应的搭档的版本
(注意,一定要一个版本进行匹配下载,不然出错的话查起来很麻烦)传送门:tensorflow官网
2、下载anaconda,这里不赘述
选择自己想下载的版本:annaconda历史下载地址
anaconda的安装步骤可参考上一篇文章,里面有详细的安装步骤
3、下载CUDA
下载地址:https://developer.nvidia.com/cuda-toolkit-archive
1)查看电脑显卡
我已经查过自己电脑的显卡是:NVIDIA GeForce940MX,大家下载Tensorflow-gpu之前也最好去查看下自己的电脑是否适配,传送门如下:网址:http://developer.nvidia.com/cuda-gpus
查看显卡及参考网址:https://developer.nvidia.com/zh-cn/cuda-gpus
2)下载对应的版本
4、下载cuDNN
安装地址:https://developer.nvidia.com/cudnn
注意对应的版本。
下载完成后,获得压缩包,解压,然后找到之前cuda安装时的位置,进行文件替换,如下图
5、配置环境变量
6、在anaconda中建立tensorflow-gpu虚拟环境
这里可以建议大家看下anaconda的使用,然后安装起来就能理解每一步是什么意思了:传送门:s初学 Python 者自学 Anaconda 的正确姿势是什么
1)进入anaconda prompt,如下图
2、新建一个虚拟环境(python版本可以自定义,我用的是python3.6,虚拟环境名称为tensorflow-gpu)
1)创建一个虚拟环境
输入:conda create --name tensorflow-gpu python=3.6 anaconda
2)然后激活该虚拟环境
输入:activate tensorflow-gpu
3)安装tensorflow-gpu
这里使用清华源安装,更快速。
因为我要装tensorflow-gpu是1.14.0的,如果你装其他版本的,可以直接把版本号输入进去,但一定要注意,版本必须按照第一版块说的匹配起来,万不可乱下载,出错的话很麻烦
pip install tensorflow-gpu==1.14.0 -i https://pypi.tuna.tsinghua.edu.cn/simple
4)如果一路下来,没出错的话,那就表示你安装成功了。
5)检查是否安装成功了
这里要注意一点,一般电脑之前都有一个python版本,而我们再装anaconda时也存在一个python,那检验的时候万不可直接wins+R操作,这样的话会出现说:tensorflow is not defined
。所以必须在anaconda prompt中分别运行下面代码:
activate tensorflow-gpu
python
import tensorflow as tf
tf.test.is_gpu_available()
若第四步输出是TRUE,表示你安装gpu版本的已成功
若出错了,不好意思,自己先看看各个版本是否是匹配的,若匹配了还有问题,那就网上搜索一下,一般都是小问题(比如pip版本需要更新,numpy版本比较低,这类问题,直接都能搜到解决方案),比如下面的问题
numpy版本不匹配,解决方法如下图:
我先输入:pip install numpy==1.16.0
,发现失败了。
于是再次输入:pip install -U numpy==1.16.0
好了,成功解决问题。这里遇到问题,不要怕,一定要多查。
3)检查的截图:
注意是在anaconda的python中运行的一下命令。