大佬们帮忙看看这代码是缺少什么库，本人是深度学习新手

可以尝试运行一下，如果缺库会提示的。