我在安装好cuda和cudnn后,使用pip install tensorflow安装的tensorflow都提示不能找到GPU,
为此怀疑默认暗转的tensorflow是不带GPU支持的。
在tensorflow官网提供了多个版本的GPU支持的windows的安装包
https://www.tensorflow.org/install/pip?hl=zh-cn#system-install
下载后再安装就可以成功安装GPU支持的tensorflow了
文章讲述了在安装CUDA和CUDNN后,使用pip安装的TensorFlow未检测到GPU。作者发现TensorFlow官网提供针对Windows的GPU版本安装包,解决了这个问题。
我在安装好cuda和cudnn后,使用pip install tensorflow安装的tensorflow都提示不能找到GPU,
为此怀疑默认暗转的tensorflow是不带GPU支持的。
在tensorflow官网提供了多个版本的GPU支持的windows的安装包
https://www.tensorflow.org/install/pip?hl=zh-cn#system-install
下载后再安装就可以成功安装GPU支持的tensorflow了
您可能感兴趣的与本文相关的镜像
PyTorch 2.5
PyTorch 是一个开源的 Python 机器学习库,基于 Torch 库,底层由 C++ 实现,应用于人工智能领域,如计算机视觉和自然语言处理
347

被折叠的 条评论
为什么被折叠?