CUDA 和 CUDAtoolkit TensorFlow 在 wsl2 ubuntu 20.04 中啟用了 gpu

Question

CUDA 和 CUDAtoolkit 有什么不同嗎？ 我使用本教程安裝了 CUDA 我也運行了示例並且它有效但是要使用 gpu 運行 tensorflow 我是否需要安裝 CUDAtoolkit 和 cudnn 以及已經存在的 CUDA。

在運行訓練作業時我收到這條消息

2023-01-01 00:03:41.818376: W tensorflow/core/common_runtime/gpu/gpu_device.cc:1934] Cannot dlopen some GPU libraries. Please make sure the missing libraries mentioned above are installed properly if you would like to use GPU. Follow the guide at https://www.tensorflow.org/install/gpu for how to download and setup the required libraries for your platform.
Skipping registering GPU devices...

Answer 1

Tensorflow對GPU的要求和配置非常講究。 您確實需要安裝 CUDNN，我相信 CudaToolkit 可能是安裝 CUDNN 的必要條件。

我建議按照官方文檔安裝 tensorflow 和 gpu 功能。 你可以在這里看到所有這些： https://www.tensorflow.org/install/pip

您還可以查看 Lambda 實驗室，它有一種非常簡化的方式來安裝 tensroflow GPU，特別是在 Ubuntu 上，並將要求添加到您的apt存儲庫以進行自動更新。 https://lambdalabs.com/lambda-stack-deep-learning-software