Tensorflow 如何知道 cuda 版本？

Question

我有一個舊的桌面，我想在中斷后使用 Tensorflow。 我卸載了以前版本的 Cuda 並安裝了 11.3。 當我嘗試使用 Tensorflow 並檢查 Cuda 包版本時，我得到 11.2，這是一個舊版本。

>>> import tensorflow as tf
2021-06-25 18:02:37.515683: W tensorflow/stream_executor/platform/default/dso_loader.cc:64] Could not load dynamic library 'libcudart.so.11.0'; dlerror: libcudart.so.11.0: cannot open shared object file: No such file or directory
2021-06-25 18:02:37.515712: I tensorflow/stream_executor/cuda/cudart_stub.cc:29] Ignore above cudart dlerror if you do not have a GPU set up on your machine.
>>> sys_details = tf.sysconfig.get_build_info()
>>> sys_details["cuda_version"]
'11.2'

Tensorflow 如何查找 Cuda 版本應該是什么？ 這些舊版本的殘余物會在哪里閑逛？

編輯1：

還有一件事： nvcc --version返回正確的版本。

Answer 1

嘗試使用此命令

export CUDA_HOME=/usr/local/cuda-11.3