簡單作法如下
1.安裝 NVIDIA 官方 driver
2.建立 conda env (TF2.2) 並一併安裝 tensorflow-gpu
這步驟會把 cuda toolkit、 CUDNN 一起安裝
3.啟動 ENV (TF2.2)
4.跑 code

—————
1.安裝 NVIDIA 官方 driver:
清除原有的nvidia driver (如果你有的話)
sudo apt-get purge nvidia*
加入顯卡 ppa
sudo add-apt-repository ppa:graphics-drivers
package 更新
sudo apt-get update
sudo apt upgrade
Unable to lock the administration directory (/var/lib/dpkg/), is another process using it?
找出目前支援的GPU driver 版本
ubuntu-drivers list
安裝 nvidia driver
sudo apt install nvidia-driver-VERSION_NUMBER_HERE
我目前 (2021/7) 使用 460 版本
sudo apt install nvidia-driver-460
安裝完後重啟
sudo reboot
檢查是否安裝成功
nvidia-smi

2.
conda create –name tf python=3.7.5 tensorflow-gpu==2.2.0
conda activate tf