Ubuntu20.04安装tensorflow2.8.0+CUDA11.4

原创

已于 2022-08-19 22:16:43 修改 · 4.2k 阅读

标签

#深度学习 #tensorflow

收录于

于 2022-04-24 23:39:07 首次发布

本文档详细介绍了如何在Ubuntu20.04上安装TensorFlow2.8.0，CUDA11.4和cuDNN。首先，创建一个Anaconda虚拟环境，然后通过pip安装TensorFlow。接着，从NVIDIA官方网站下载并安装CUDA，同时更新环境变量。对于cuDNN，提供了手动安装和deb包安装两种方法。最后，通过代码测试验证CUDA和TensorFlowGPU版的安装成功。

Ubuntu20.04安装tensorflow2.8.0+CUDA11.4

需要事先安装好Anaconda

1. 创建虚拟环境

# python 版本可选
conda create -n tensorflow python=3.7.11
conda activate tensorflow

2. 安装tensorflow

我安装时tensorflow-2.8.0就是最新版本，默认直接安装了

# 我安装时tensorflow-2.8.0就是最新版本，默认直接安装了
pip install --upgrade tensorflow

3. 安装CUDA

下载地址：https://developer.nvidia.com/cuda-toolkit-archive
常用的安装方式有deb和runfile，官方下面都给出了相应的命令，我比较习惯用runfile。另外安装过程中可能会提示卸载和以前cuda相关的文件，这个直接同意，另外默认是安装配套的NVIDIA驱动和CUDA的示例，不过我已经