Tensorflow安装及配置GPU
Tensorflow安装及配置GPU
安装Tensorflow
CPU版本
安装：pip install tensorflow
GPU版本
安装GPU版本必须有GPU硬件的支持，Tensorflow对NVIDIA显卡的支持较为完备。AMD的显卡也开始对Tensorflow提供支持。
安装：pip install tensorflow-gpu
注：tensorflow2.0不支持MAC系统，所以mac下只能安装cpu版本。
配置GPU环境
安装NVIDIA相关软件：
- NVIDIA驱动程序(需要410.x或更高版本)
- CUDA(需要10.0，推荐10.0.130，不可以是10.1)
- cudnn(需要大于7.4.1，推荐7.6.0)
推荐使用conda来安装CUDA和CUDNN，因为这样的安装方式不需要再设置环境变量
查看GPU信息
命令行中运行：nvidia-smi
要是没有该程序没有在环境变量中的话，可以在驱动安装路径下找到
安装CUDA
conda install cudatoolkit=10.0.130
安装cudnn
conda install cudnn=7.6.0
安装顺序
可能要先装cudnn再装CUDA，因为conda会自己判断版本，有的时候可能会把CUDA降级。
确认GPU是否安装成功
tf.test.is_gpu_available()
成功的话会返回Ture