☰

支持 GPU 的 Tensor Flow

tag-icon • tag-icon tensorflow
支持 GPU 的 Tensor Flow 
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'))"

相关内容