tensorflow2 安装 | ubuntu 20.04, cuda 12.2, RTX 4090

各种版本：

OS: Ubuntu 20.04

GPU: RTX4090

Cuda: 12.2

Driver Version: 535.183.01

1. 检查对应的 tensorflow version

Tensorflow与Python、CUDA、cuDNN的版本对应表

选择：tensorflow-2.15.0, python 3.9, cudnn 8.9.

tf 2.13.0及以后的版本都不需要额外安装tensorflow-gpu

2. 安装tf==2.15.0

conda create -n tf2_env python=3.9

conda activate tf2_env

pip install tensorflow==2.15.0

python -c "import tensorflow as tf; print(tf.__version__)"

3.安装cudnn==8.9

下载cudnn；

Download cuDNN v8.9.7 (