ubuntu系统使用Anaconda安装tensorflow-gpu环境_ubuntu安装tensorflow-gpu

ubuntu系统使用Anaconda安装tensorflow-gpu环境

本次安装的系统版本为Ubuntu 22.04。安装下来个人感觉是主要关注python版本和CUDA驱动版本，注意显卡驱动可以向下兼容，其他的问题不大。

下面记一下我的具体安装过程：

1. 首先准备好要安装的Python环境，推荐使用Anaconda创建的虚拟环境，我的Python版本为3.8。
2. 安装tensorflow-gpu：

pip install tensorflow-gpu==2.2.0 -i https://pypi.tuna.tsinghua.edu.cn/simple
1

3. 安装CUDA工具包：

conda install cudatoolkit=10.1 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/linux-64/

1
2

4. 安装cuDNN：

conda install cudnn=7.6.5
1

5. 验证环境是否配置成功：

import tensorflow as tf
# 检查tensorflow是否得到CUDA支持，安装成功则显示true，否则为false
tf.test.is_built_with_cuda()
# 检查tensorflow是否可以获取到GPU，安装成功则显示true，否则为false
tf.test.is_gpu_available()
1
2
3
4
5

注意事项：如果报错numpy没有object的错误，更改相应的numpy版本就可以解决。