Ubuntu18.04安装python3.8+pytorchV1.7.1 + cuda11.0+cuDNN 8.0.0 + tensorflow2.4.0+tensorflow_gpu-2.4.0

上一篇是在Windows下配置的环境，代码太占显存，换一个服务器。呜呜呜呜呜呜，又要配环境！！！！！这一篇在Ubuntu下从头到尾配置。

设备：Ubuntu18.04+2*Titan RTX（24GB）

一、安装NVIDIA驱动（终端出现下图结果表示安装成功）。驱动版本为515.48.07，CUDA版本为 11.7。我装好cudnn后发现tensorflow的gpu版本还不支持cuda11.7！！！啊啊啊啊啊啊啊啊啊啊啊啊啊又是疯掉的一早上。

参考：ubuntu18.04 安装Nvidia驱动的三种方式（必看）_低吟浅笑的博客-CSDN博客_ubuntu18.04安装nvidia显卡驱动

二、查看对应版本！！！！

tensorflow目前最高支持cuda11.2。

1.首先查看NVIDIA对应的CUDA版本。Release Notes :: CUDA Toolkit Documentation (nvidia.com)