Linux tensorflow-gpu==2.5.0 环境搭建(RTX 3080 Ubantu)

目录

          环境配置

Miniconda3安装

tensorflow-gpu安装

---

本人配置：RTX 3080 Ubantu

环境配置：

tensorflow-gpu=2.5.0

cudatoolkit=11.3.1

cudnn=8.2.1

Miniconda3安装

wget https://mirrors.tuna.tsinghua.edu.cn/anaconda/miniconda/Miniconda3-py39_4.12.0-Linux-x86_64.sh

bash Miniconda3-py38_4.12.0-Linux-x86_64.sh （找到安装包位置，cd进入位置再bash）

关闭或重启一个终端(更新数据)

tensorflow-gpu安装

创建虚拟环境：

conda create -n tf2.5 python==3.9

进入pycharm，更换环境为conda环境

路径为：setting->Project->Python interpreter->

Add interpreter>Conda Environment->interpreter->/miniconda3/envs/tf2.5/bin/python3

进入虚拟环境： conda activate tf2.5

安装tensotflow-gpu和tensorflow

(用pip安装是因为conda官网上没有对应的gpu版本,全程都要在ubantu终端并且在tf2.5的环境下使用命令)

conda install tensorflow==2.5.0

pip install tensorflow-gpu==2.5.0

pip install keras Image matplotlib tqdm opencv-python

安装cudnn和cudnn：

(根据自己安装tensorflow-gpu版本对应的cuda和cudnn，我自己是安装的如下版本)

conda install cudatoolkit==11.3.1 cudnn==8.2.1

成功结果如下图：