Tensorflow_gpu软件安装(windowns开发环境搭建):
在安装之前请确认你的电脑是否具备CUDA支持,详情可以到官网去查询
在设备管理器里面复制你的GPU显卡版本号
搜索软件搜索”你的GPU型号+SPECIFICATION”
如果不支持你就只能安装cpu版本的了。
在这里我的电脑是支持的。
本教程将基于 Anaconda 安装 TensorFlow。
1.Anaconda安装:
首先需要下载 Anaconda 下载网址可以到官网或则国内镜像站去下载,以下我分享两个下载地址
官网下载:https://www.anaconda.com/distribution/
国内清华大学开源镜像站:https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/ 选择对应需求版本即可。
我安装的如下版本。
建议使用第二个网站下载,下载飞快,下载完成后点击exe运行会出现如下界面
余下步骤默认即可。
验证Anaconda是否安装成功的方法:
命令窗口中输入“conda --version” ----->得到conda 4.8.2
看到了这个结果,恭喜你,你已经成功的安装上了Anaconda了,那么我们继续。
2.CUDA安装:
请看清楚你需要的TENSORFLOW型号,以及它需要的环境!
这里我准备安装的是Tensorflow2.0的版本,选择CUDA10的版本,不太清楚CUDA10.1和CUDA10.2的版本是否可以,网上据说有问题,本人在这里直接使用的CUDA10.0的版本。
下载完成后运行安装即可
检查安装是否成功打开cmd输入nvcc -V
出现如上界面即为安装成功。
如果出现安装不成功的问题尝试卸载
然后删除C:Program Files文件夹下的NVIDIA GPU Computing Toolkit文件
此时打开设备管理器
能正常识别设备后开始重新安装CUDA
3.CUDNN安装:
到官网下载对应的cuDNN版本,注意一定要和cuda配对的,按照上面的图片我们安装cudnn7.4的版本,官网下载需要先注册账号
解压下载后的文件,并点击进去复制里面的文件
如果的cuda是默认安装的话找到这个文件夹
C:Program FilesNVIDIA GPU Computing ToolkitCUDAv10.0
在这一级目录直接粘贴就安装好了
4.Tensorflow安装:
- 在ANACONDA里创建名为TENSORFLOW的环境(你可以叫他任何名字,这里我叫这个环境为TENSORFLOW。我使用python3.7
conda create -n tensorflow pip python=3.7
- 看到如下界面后按提示输入
activate tensorflow激活该环境
- 继续输入以下命名安装tensorflow:
pip install --ignore-installed --upgrade tensorflow-gpu
假如你想安装某个特定版本的tensorflow,可以输入如下命令,根据自己的喜好,替代那个2.0.0即可:pip install --ignore-installed --upgrade tensorflow_gpu==2.0.0
检测是否安装成功,打开如下
输入:activate tensorflow
然后输入:python
再然后输入:
>>> import tensorflow as tf
>>> print(tf.__version__)