关于我装的GPU版tensorflow然后却是cpu在跑这件事
神经网络的层数越来越多,程序跑的也是越来越慢,打开任务管理器发现我的GPU根本没用到,白瞎我买的电脑,和费那劲装的tensorflowGPU版
首先看一下GPU是否可用
import tensorflow as tf
print(tf.config.list_physical_devices('GPU'))
我的输出是[],也就是空
在cmd窗口激活我们用的tensorflow环境,会显示找不到cusolver64_10.dll,我们进入cuda的目录,我的是默认的C:\Program Files\NV
原创
2021-11-19 17:23:43 ·
3957 阅读 ·
0 评论