在链接到 GPU 的新 Ubuntu 22.04 上安装 TensorFlow

Question

我一整天都在尝试在装有 Ubuntu 22.04 的新机器上安装支持 GPU 的 TensorFlow，但至少可以说我正在努力挣扎。

安装是全新的，我的机器上有可用的 Nvidia RTX 6000，但是我一生都无法让张量流接受它。

我运行以下命令（来自全新版本）：

#General system updates
sudo apt update
apt install python3-pip

#Install the nvidia driver
sudo apt install nvidia-driver-550 nvidia-dkms-550

#Install CUDA
apt install nvidia-cuda-toolkit


#Install Bazelisk
wget https://github.com/bazelbuild/bazelisk/releases/download/v1.8.1/bazelisk-linux-amd64
chmod +x bazelisk-linux-amd64
sudo mv bazelisk-linux-amd64 /usr/local/bin/bazel

#Install tensorflow
git clone https://github.com/tensorflow/tensorflow.git
cd tensorflow
./configure

但是这会遇到以下错误

Inconsistent CUDA toolkit path: /usr vs /usr/lib
Asking for detailed CUDA configuration...

我根本无法修复这个问题 - 我已经尝试过重建机器 5 次以及阳光下的其他一切，但我一生都无法让它工作。

非常感谢任何和所有的帮助。

Answer 1

对于每个和我有同样问题的人来说 - 它可以非常容易地解决。无需从源代码构建，只需使用 pip (https://www.tensorflow.org/install/pip)，但要下载以前的版本，特别是

pip install tensorflow[and-cuda]==2.15.1

通过上述内容，它可以与 pip 完美配合，而无需从源代码构建