
tensorflow
文章平均质量分 56
归海听雪
这个作者很懒,什么都没留下…
展开
-
基于docker下部署深度学习环境(GPU、tensorrt、tensorflow)
一、系统环境Ubuntu 18 4.15.0-159-generic二、显卡驱动安装1、查看显卡型号以及驱动#执行命令ubuntu-drivers devices#显示结果WARNING:root:_pkg_get_support nvidia-driver-390: package has invalid Support Legacyheader, cannot determine support level== /sys/devices/pci0000:00/00.原创 2022-01-11 11:48:56 · 1289 阅读 · 0 评论 -
解决GPU显存无法释放问题
经常有开发反馈他们的程序已停掉,但是GPU显存无法释放,我们在使用tensorflow+pycharm 或者PyTorch写程序的时候, 有时候会在控制台终止掉正在运行的程序,但是有时候程序已经结束了,nvidia-smi也看到没有程序了,但是GPU的内存并没有释放,这是怎么回事呢?使用PyTorch设置多线程(threads)进行数据读取(DataLoader),其实是假的多线程,他是开了N个子进程(PID都连着)进行模拟多线程工作,所以你的程序跑完或者中途kill掉主进程的话,子进程的GPU显存并不会原创 2021-12-06 14:32:07 · 10661 阅读 · 2 评论