开发者学堂课程【Tensorflow2.0入门与实战:Tensorflow 2.0 GPU 版本的安装】学习笔记,与课程紧密联系,让用户快速学习知识。
课程地址:https://developer.aliyun.com/learning/course/664/detail/11097
Tensorflow 2.0 GPU版本的安装
内容介绍:
一、安装要求
二、如何查看驱动版本
三、搭建开发环境 Visual C++ 安装
四、Tensorflow 2.0 GPU 版本安装步骤
一、安装要求
1.安装 GPU 版本必须有 GPU 硬件的支持。
2.TensorFlow 对 NVIDIA 显卡的支持较为完备。
3.使用 conda 来安装GPU版本,对于 NVIDIA 显卡,要求其 CUDA Compute Capability 须不低于3.0。
算力参考: https://developer.nvidia.com/cuda-gpus
4.驱动版本注意: NVIDIA 驱动程序需410.x 或更高版本。
5.以下两个是依赖库的版本要求,不需要手动安装。
CUDA 的版本需要是10.0
Cudnn 版本号需要不小于7.4.1
二、如何查看驱动版本
NVIDIA 驱动程序版本查看:命令行中执行: nvidia-smi
如下图,
nvidia-smi 满足大于410,所以可以安装 Tensorflow 2.0。
如果驱动版本不能达到安装 Tensorflow 2.0 的要求,可通过下面的地址下载。
NVIDIA 驱动程序下载地址:
https://www.nvidia.com/Download/index.aspx?lang=en-us
三、搭建开发环境Visual C++安装
Visual C++安装,可从微软的官网下载
https://support.microsoft.com/zh-cn/help/2977003/the-latest-supported-visual-c-downloads(双击安装,安装完需重启计算机)
四、Tensorflow 2.0 GPU 版本安装步骤
1.安装 Miniconda
请注意:如果您之前安装过 Anaconda ,请务必卸载。
注意:此处最新版本仅指当前的最新版本,2020.5前,后面如果 miniconda 新版本默认使用了3.8就不合适了。
下载完成后如下图,双击打开安装默认指示安装即可
2.安装 Miniconda
设置 conaa 使用国内源
3.安装GPU版本:
conda install tensorflow-gpu==2.0.0
完成
4.检验安装是否完成
在搜索框中搜索 anaconda prompt 进入 anaconda prompt 的命令行,命令行中输入 jupyter notbook 打开 jupyter,然后建立一个 notbook 文件。
如上图,如果 c 出现 ture 表示安装成功。