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