Tensorflow 2.14.0(with cuda)未注册CUDA？

brgchamk 于 6个月前发布在其他

关注(0)|答案(1)|浏览(91)

我尝试在GPU上使用Tensorflow。我的系统是Fedora Linux 38，NVIDIA驱动程序535.113.01（当前最新）在我的系统上按预期工作。
我用Python 3.9创建了一个Python环境（afaik 3.11版无法使用pip安装TF 2.14.0）。
在激活的环境中，我安装TF如下：

python3.9 -m pip install tensorflow[and-cuda]

字符串
我可以看到pip安装了Tensoflow和许多必需的库（cuds，cuda，cull和其他）。
一切看起来都很好，但是当我导入tensorflow时，我得到了这个错误：
第一个月
然后，我使用了来自Tensorflow：tensorflow/tensorflow：latest-gpu的docker镜像作为最后一个选项，但这显示了完全相同的错误
其他信息

Python版本：版本应该是>=3.9和< 3.11（我尝试了许多版本之间）
没有CUDA/CUdNN的本地安装：我删除了CUDA & CUdNN，使用“查找/ -iname cuda”，以确保没有痕迹
我重新安装了驱动程序，并尝试了版本520太，同样的问题

我试图弄清楚发生了什么，当Tensorflow说“......一个已经在2023-10-22注册”为cuDNN，cuBLAS，cuFFT，以便能够至少了解要查找的内容，但我在网上找不到任何有用的东西。

tensorflow

来源：https://stackoverflow.com/questions/77338229/tensorflow-2-14-0-with-cuda-not-registering-cuda

1条答案

按热度按时间

zqry0prt1#

经过72 hs的尝试和错误没有运气，包括重新安装各种版本的nvidia驱动程序，我重新安装了操作系统，并遵循了以下过程：

安装驱动程序（没有cuda！）
创建python环境
安装tensorflow[and-cuda]（使用Python 3.9

现在导入Tensorflow，虽然它显示相同的警告，但将使用GPU：

physical_devices = tf.config.experimental.list_physical_devices('GPU')
if len(physical_devices) > 0:
    print("We got a GPU")
    tf.config.experimental.set_memory_growth(physical_devices[0], True)
else:
    print("Sorry, no GPU for you...")

字符串
这表明我终于有我的GPU启动和运行.现在我可以去有另一个72 hs的摔跤与数据.

赞(0）回复(0）举报 6个月前

我来回答

Tensorflow 2.14.0(with cuda)未注册CUDA？

1条答案

相关问题

热门标签

最新问答