CohenV
CohenV
大哥，我根据你的指引在linux搞好了。能运行，但是tensorflow识别不到cuda，用不了GPU。 你改TensorFlow的版本了吗？
> > 大哥，我根据你的指引在linux搞好了。能运行，但是tensorflow识别不到cuda，用不了GPU。 你改TensorFlow的版本了吗？ > > ``` > import torch > > if torch.cuda.is_available(): > print("CUDA is available. Training on GPU.") > else: > print("CUDA is not available. Training on...
> > 大哥，我根据你的指引在linux搞好了。能运行，但是tensorflow识别不到cuda，用不了GPU。 你改TensorFlow的版本了吗？ > > 我本地的： tensorflow 2.12.0 tensorflow-estimator 2.12.0 tensorflow-intel 2.12.0 tensorflow-io-gcs-filesystem 0.31.0 谢谢回复，可以看看你项目虚拟环境中完整的包和版本列表吗？
❯ nvidia-smi Fri May 30 10:50:56 2025 +-----------------------------------------------------------------------------------------+ | NVIDIA-SMI 570.144 Driver Version: 570.144 CUDA Version: 12.8 | |-----------------------------------------+------------------------+----------------------+ | GPU Name Persistence-M | Bus-Id Disp.A | Volatile Uncorr. ECC...
大哥，你的cudnn版本是什么？
解决了，cudnn更新到9.0就好了。 谢谢各位的回复