镜像版本:dzhwinter/benchmark:latest;
CPU版本测试,环境变量设置
export CUDA_VISIBLE_DEVICES=;
export OMP_NUM_THREADS=1
运行命令:python stacked_dynamic_lstm.py --device CPU
输出信息如下:
Reading data to memory
Done. data size 24996
Traceback (most recent call last):
File "stacked_dynamic_lstm.py", line 206, in
main()
File "stacked_dynamic_lstm.py", line 187, in main
train_loop(args.pass_num, args.crop_size)
File "stacked_dynamic_lstm.py", line 174, in train_loop
tensor_words = to_lodtensor([x[0] for x in data], place)
File "stacked_dynamic_lstm.py", line 201, in to_lodtensor
res.set_lod([lod])
MemoryError: std::bad_alloc
镜像版本:dzhwinter/benchmark:latest;
CPU版本测试,环境变量设置
export CUDA_VISIBLE_DEVICES=;
export OMP_NUM_THREADS=1
运行命令:python stacked_dynamic_lstm.py --device CPU
输出信息如下:
Reading data to memory
Done. data size 24996
Traceback (most recent call last):
File "stacked_dynamic_lstm.py", line 206, in
main()
File "stacked_dynamic_lstm.py", line 187, in main
train_loop(args.pass_num, args.crop_size)
File "stacked_dynamic_lstm.py", line 174, in train_loop
tensor_words = to_lodtensor([x[0] for x in data], place)
File "stacked_dynamic_lstm.py", line 201, in to_lodtensor
res.set_lod([lod])
MemoryError: std::bad_alloc