python 如何让tensorflow使用GPU?

vngu2lb8  于 2023-02-18  发布在  Python
关注(0)|答案(1)|浏览(213)

我正在使用python,我想在我的GTX 2080 TI上使用tensorflow,但是tensorflow只使用CPU。
当我请求计算机上设备时,它总是返回一个空列表:

In [3]: tf.config.list_physical_devices('GPU')
Out[3]: []

我试试这个帖子:How do I use TensorFlow GPU?,但我不使用cuda和tennsorflow-gpu似乎过时了。
我也尝试了这个做得很好的教程https://www.youtube.com/watch?v=hHWkvEcDBO0没有成功。
我安装了卡驱动程序,CUDA和CudNN,但我仍然得到同样的问题。我还卸载了TensorFlow和Keras,并再次安装它们,但没有成功。
我不知道怎么才能找到丢失的东西,或者我是否做错了什么。
Python 3.10张力流版本:2.11.0 Cuda版本:11.2 cudNN:8.1
这行代码告诉我cuda没有构建:

print(tf_build_info.build_info) 
OrderedDict([('is_cuda_build', False), ('is_rocm_build', False), ('is_tensorrt_build', False), ('msvcp_dll_names', 'msvcp140.dll,msvcp140_1.dll')])
lstz6jyr

lstz6jyr1#

从TensorFlow 2.11开始,本机窗口不再支持GPU。您可以选择使用Direct ML插件教程here

相关问题