神经网络的层数越来越多,程序跑的也是越来越慢,打开任务管理器发现我的GPU根本没用到,白瞎我买的电脑,和费那劲装的tensorflowGPU版
首先看一下GPU是否可用
import tensorflow as tf
print(tf.config.list_physical_devices('GPU'))
我的输出是[],也就是空
在cmd窗口激活我们用的tensorflow环境,会显示找不到cusolver64_10.dll,我们进入cuda的目录,我的是默认的C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.1\bin
,把cusolver64_11.dll改名改成10
到这里我就看到我的GPU了 [PhysicalDevice(name=’/physical_device:GPU:0’, device_type=‘GPU’)]
继续运行程序,我以前可以正常运行的代码开始报这个错
NotFoundError: No algorithm worked! [[node vgg_net/conv2d/Conv2D
(defined at :94) ]]
[Op:__inference_train_function_3048]E