AntaresWendi的博客

原创 GPU占用死循环

import os# os.environ["CUDA_VISIBLE_DEVICES"] = "1" #指定第1块gpuimport torchdata = torch.zeros(4000,1000,1000).cuda()while True: a=1tensor 大小可以根据显存大小设置，（4000，1000，1000），可以占用16G。命令行直接运行，CUDA_VISIBLE_DEVICES=1 python zhanyong.py如果是使用VS.