![支持 GPU 的 Tensor Flow](https://linux22.com/image/951676/%E6%94%AF%E6%8C%81%20GPU%20%E7%9A%84%20Tensor%20Flow.png)
TUF-Gaming-FX505DT-FX505DT:/home/jithin# lspci | grep VGA
01:00.0 VGA compatible controller: NVIDIA Corporation TU117M [GeForce GTX 1650 Mobile / Max-Q] (rev ff)
05:00.0 VGA compatible controller: Advanced Micro Devices, Inc. [AMD/ATI] Picasso/Raven 2 [Radeon Vega Series / Radeon Vega Mobile Series] (rev c2)
我最近订购了一台 gtx 3060 + R5 7600x 系统,它将在 1-2 周内到达,我想在我的笔记本电脑上尝试 tensorflow,我目前有一台华硕 tuf fx 505DT 笔记本电脑
我正在尝试安装支持 gpu 的 tensorflow,但在选择 cuda、cudnn、gcc 版本以及如何安装它们时遇到很多问题。我在 ubuntu 22.04 上。有人可以发布所有步骤,包括要安装的版本吗?
答案1
我使用了以下命令序列并且成功了 - 在支持 GPU(RTX 3080ti 和 RTX 2060)的 Ubuntu 22.04 上安装 TensorFlow-2.11.0
sudo apt update
sudo apt upgrade
nvcc --version
sudo apt install nvidia-cuda-toolkit
sudo apt install python3-dev python3-pip
sudo apt install python3-pip
pip3 install --upgrade pip
nvcc --version
nvidia-smi
pip3 --version
sudo pip3 install nvidia-cudnn-cu11==8.5.0.96
sudo pip3 install tensorflow==2.11.0
sudo apt install nvidia-cudnn
python3 -c "import tensorflow as tf; print(tf.config.list_physical_devices('GPU'))"