Spaces:
Running
on
Zero
Running
on
Zero
File size: 442 Bytes
17cd746 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 |
# -*- coding: utf-8 -*-
import os
import time
import torch
import argparse
parser = argparse.ArgumentParser(description='inf')
parser.add_argument('--gpu', default='1', type=str, help='index of gpu to use')
args = parser.parse_args()
os.environ["CUDA_VISIBLE_DEVICES"] = args.gpu
n = 1000
x = torch.zeros(4, n, n).cuda()
rest_time = 0.0000000000001
while True:
y = x * x
time.sleep(rest_time)
y1 = x * x
|