集群服务器如何安装tensorflow-gpu

1. 未安装cuda需要先安装cuda--命令行：sudo apt install nvidia-cuda-toolkit

输入该命令行之后一般要求输入权限密码.

ps: 不清楚是否安装可输入nvcc -V来检查，如出现以下图识即表明cuda已安装成功/

2. 创建一个新环境

conda create -n xxx(命名) python=3.x(你想要的版本)

3.激活环境

conda activate xxx(刚刚的命名)

4.安装tensorflow-gpu环境

pip install tensorflow-gpu==2.x(我这里安装2.6，安装其他版本与自己的cuda版本对应即可)

5.安装cuda以及cudnn

conda install cudatoolkit=11.3

conda install cudnn=8.2

（以上版本是需要跟你安装的tensorflow版本对应的）

6.检查安装情况

--python

--import tensorflow as tf

如果出现如下错误：

 

即需要输入命令行：pip install protobuf==3.20

--如果没有出现错误：

即安装成功