GPU占用死循环

import os
# os.environ["CUDA_VISIBLE_DEVICES"] = "1"  #指定第1块gpu

import torch
data = torch.zeros(4000,1000,1000).cuda()

while True:
    
    a=1

tensor 大小可以根据显存大小设置，（4000，1000，1000），可以占用16G。

命令行直接运行，

CUDA_VISIBLE_DEVICES=1 python zhanyong.py

如果是使用VScode，不想占用终端的话，可以使用tmux命令放到后台，即使VScode关掉了，也会在占用。

新建后台任务zhanyong1，

tmux new -s zhanyong1

运行占用程序，crtl b + d 退出后台。查看后台任务，

tmux a -t zhanyong1