ASRT_SpeechRecognition
ASRT_SpeechRecognition copied to clipboard
关于模型训练报错问题
模型在训练时,保存一次模型之后会出现如下错误。本人电脑显卡为GeForce 1060,batch_size设置为8,请问是因为显卡内存不够的原因吗?
2020-06-08 21:06:33.504024: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1618] Found device 0 with properties:
name: GeForce GTX 1060 6GB major: 6 minor: 1 memoryClockRate(GHz): 1.7845
pciBusID: 0000:01:00.0
2020-06-08 21:06:33.504241: I tensorflow/stream_executor/platform/default/dso_loader.cc:44] Successfully opened dynamic library cudart64_100.dll
2020-06-08 21:06:33.504373: I tensorflow/stream_executor/platform/default/dso_loader.cc:44] Successfully opened dynamic library cublas64_100.dll
2020-06-08 21:06:33.504513: I tensorflow/stream_executor/platform/default/dso_loader.cc:44] Successfully opened dynamic library cufft64_100.dll
2020-06-08 21:06:33.504648: I tensorflow/stream_executor/platform/default/dso_loader.cc:44] Successfully opened dynamic library curand64_100.dll
2020-06-08 21:06:33.504782: I tensorflow/stream_executor/platform/default/dso_loader.cc:44] Successfully opened dynamic library cusolver64_100.dll
2020-06-08 21:06:33.504917: I tensorflow/stream_executor/platform/default/dso_loader.cc:44] Successfully opened dynamic library cusparse64_100.dll
2020-06-08 21:06:33.505052: I tensorflow/stream_executor/platform/default/dso_loader.cc:44] Successfully opened dynamic library cudnn64_7.dll
2020-06-08 21:06:33.505494: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1746] Adding visible gpu devices: 0
2020-06-08 21:06:33.505660: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1159] Device interconnect StreamExecutor with strength 1 edge matrix:
2020-06-08 21:06:33.505794: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1165] 0
2020-06-08 21:06:33.505878: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1178] 0: N
2020-06-08 21:06:33.506282: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1304] Created TensorFlow device (/job:localhost/replica:0/task:0/device:GPU:0 with 4708 MB memory) -> physical GPU (device: 0, name: GeForce GTX 1060 6GB, pci bus id: 0000:01:00.0, compute capability: 6.1)
Test Count: 0 / 4
2020-06-08 21:06:33.629067: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1618] Found device 0 with properties:
name: GeForce GTX 1060 6GB major: 6 minor: 1 memoryClockRate(GHz): 1.7845
pciBusID: 0000:01:00.0
2020-06-08 21:06:33.629275: I tensorflow/stream_executor/platform/default/dso_loader.cc:44] Successfully opened dynamic library cudart64_100.dll
2020-06-08 21:06:33.629405: I tensorflow/stream_executor/platform/default/dso_loader.cc:44] Successfully opened dynamic library cublas64_100.dll
2020-06-08 21:06:33.629613: I tensorflow/stream_executor/platform/default/dso_loader.cc:44] Successfully opened dynamic library cufft64_100.dll
2020-06-08 21:06:33.629743: I tensorflow/stream_executor/platform/default/dso_loader.cc:44] Successfully opened dynamic library curand64_100.dll
2020-06-08 21:06:33.629874: I tensorflow/stream_executor/platform/default/dso_loader.cc:44] Successfully opened dynamic library cusolver64_100.dll
2020-06-08 21:06:33.630009: I tensorflow/stream_executor/platform/default/dso_loader.cc:44] Successfully opened dynamic library cusparse64_100.dll
2020-06-08 21:06:33.630141: I tensorflow/stream_executor/platform/default/dso_loader.cc:44] Successfully opened dynamic library cudnn64_7.dll
2020-06-08 21:06:33.630637: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1746] Adding visible gpu devices: 0
2020-06-08 21:06:33.630937: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1159] Device interconnect StreamExecutor with strength 1 edge matrix:
2020-06-08 21:06:33.631069: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1165] 0
2020-06-08 21:06:33.631151: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1178] 0: N
2020-06-08 21:06:33.631568: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1304] Created TensorFlow device (/job:localhost/replica:0/task:0/device:GPU:0 with 4708 MB memory) -> physical GPU (device: 0, name: GeForce GTX 1060 6GB, pci bus id: 0000:01:00.0, compute capability: 6.1)
Traceback (most recent call last):
File "D:/语音数据集/ASRT_SpeechRecognition-master/train_mspeech.py", line 48, in
Process finished with exit code 1
关于这个问题可以看我的项目文档,1060的显存只有6GB,应该是不够用的 https://asrt.ailemon.me/docs/issues
谢谢,我看一下
------------------ 原始邮件 ------------------ 发件人: "nl8590687"<[email protected]>; 发送时间: 2020年6月8日(星期一) 晚上9:52 收件人: "nl8590687/ASRT_SpeechRecognition"<[email protected]>; 抄送: "846738812"<[email protected]>;"Author"<[email protected]>; 主题: Re: [nl8590687/ASRT_SpeechRecognition] 关于模型训练报错问题 (#193)
关于这个问题可以看我的项目文档,1060的显存只有6GB,应该是不够用的 https://asrt.ailemon.me/docs/issues
— You are receiving this because you authored the thread. Reply to this email directly, view it on GitHub, or unsubscribe.