我正在尝试使用tensorflow nightly 2.12（以便能够使用Cuda 12.0）在GPU（NVIDIA GeForce RTX 3050）上运行一些模型。我遇到的问题是，显然我所做的每项检查似乎都是正确的，但最终脚本无法检测到 GPU。我花了很多时间试图了解正在发生的事情，但似乎没有任何效果，因此任何建议或解决方案都将受到欢迎。GPU 似乎正在为 torch 工作，正如您在问题的最后看到的那样。
我将展示我所做的一些有关 CUDA 的最常见检查（Visual Studio Code 终端），希望您发现它们有用：
检查CUDA版本：
$ nvcc --version
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2023 NVIDIA Corporation
Built on Fri_Jan__6_16:45:21_PST_2023
Cuda compilation tools, release 12.0, V12.0.140
Build cuda_12.0.r12.0/compiler.32267302_0
检查与CUDA库的连接是否正确：
$ echo $LD_LIBRARY_PATH
/usr/cuda/lib
检查 GPU …