Tensorflow2.4环境配置

ubuntu 18.04 下 tensorflow 环境配置
1. 显卡要求
显卡计算能力要求 capability>3.5
查询:https://developer.nvidia.com/zh-cn/cuda-gpus
Tensorflow2.4环境配置_第1张图片
Tensorflow2.4环境配置_第2张图片
Tensorflow2.4环境配置_第3张图片
Tensorflow2.4环境配置_第4张图片
2.驱动程序

驱动程序对应的cuda驱动>=CUDA Version 11.0,最好11.2
通过nvidia-smi查看驱动版本,版本过低需要升级
Cudatoolkit版本11.0,Cudnn 8.0

Conda install cudatoolkit=11.0 cudnn

如果直接用conda查看可下载的版本最高只有7.6.5
从官网下载对应的cudnn8.0文件
解压下载文件

tar -zxvf cudnn-10.1-linux-x64-v7.6.0.64.tgz

将解压后cuda/lib文件夹下所有内容放置在配置环境的lib文件下
如我这里的 /anaconda3/envs/tf24/lib

3. tensorflow2.4 安装

进入环境后
pip install tensorflow-gpu==2.4.0 -i https://pypi.douban.com/simple/

你可能感兴趣的:(tensorflow,深度学习)