今天在tensorflow,一直检测不到gpu,网上查了一下,好像Windows平台不再支持 CUDA 构建。

原为链接:Tensorflow与Python、CUDA、cuDNN的版本对应表_tensorflow版本对应-CSDN博客

于是就在b站看到了一个别的方法,安装tensorflow2.6。

步骤:

1.

2.

3.

4.numpy的版本要特别注意

5.以下的版本也要注意

6.安装完验证结果:

python

import tensorflow as tf
print("TensorFlow Version:", tf.__version__)
print("Available devices: ", tf.config.experimental.list_physical_devices())
print("Num GPUs Available: ", len(tf.config.experimental.list_physical_devices('GPU')))

结果如下:

Logo

有“AI”的1024 = 2048,欢迎大家加入2048 AI社区

更多推荐