昨天遇到了一个贼傻叉的问题,跑yolov3的代码,在笔记本上运行的好好的,改到台式机上怎么都不行,读取网络权重也能读出来,但是一跑到sess.sun(…)就会发生内存全被吃满,莫名其妙的问题,现在还是不理解为什么会这样,然后就试着调了一下,怎么都不行,后来改了一下用的gpu号,平时默认都会使用0号,我给改成1了,然后居然好使了,wtf!这是咋回事啊…
修改版本号
os.environ["CUDA_VISIBLE_DEVICES"] = "1
要是有哪位大哥看到了,指点一下!!!
昨天遇到了一个贼傻叉的问题,跑yolov3的代码,在笔记本上运行的好好的,改到台式机上怎么都不行,读取网络权重也能读出来,但是一跑到sess.sun(…)就会发生内存全被吃满,莫名其妙的问题,现在还是不理解为什么会这样,然后就试着调了一下,怎么都不行,后来改了一下用的gpu号,平时默认都会使用0号,我给改成1了,然后居然好使了,wtf!这是咋回事啊…
修改版本号
os.environ["CUDA_VISIBLE_DEVICES"] = "1
要是有哪位大哥看到了,指点一下!!!