大家好，又见面了，我是你们的朋友全栈君。
【说在前面】
复杂的机器学习程序，特别是已经训练好跑的时候出的问题，不是说改CUDA就能改的。因为很多参数已经在另一个gpu里面了，单纯改一个test的代码是不够的。
如果是上述情况，只有查运行情况了，看看到底是什么程序占用了那么那么多的内存，可以的话kill掉就行。
报错如下
思路
简洁明了，他已经告诉你了，默认使用的那gpu内存不足。
在操作系统输入如下，查一下memory现在的状态：
nvidia-smi
害，发现GPU-0有一个进程正在执行导致1GB剩余都不够。
我们用GPU-1执行就行啦！
问题解决
python文件中：
import os
os.environ["CUDA_VISIBLE_DEVICES"] = '1'
解决了。
版权声明：本文内容由互联网用户自发贡献，该文观点仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容， 请发送邮件至 举报，一经查实，本站将立刻删除。
发布者：全栈程序员-用户IM，转载请注明出处：https://javaforall.cn/127682.html原文链接：https://javaforall.cn
【正版授权，激活自己账号】： Jetbrains全家桶Ide使用，1年售后保障，每天仅需1毛
【官方授权 正版激活】： 官方授权 正版激活 支持Jetbrains家族下所有IDE 使用个人JB账号...