더러운 python 해결책 모음

• torch 프린트 옵션

torch.set_printoptions(sci_mode=False)

• gpu 메모리 먹는 아가 찾기

import torch
import gc
tensors = []
tensor_sizes = []
# torch tensor objectを全部集める
for obj in gc.get_objects():
   try:
       if torch.is_tensor(obj) or (hasattr(obj, 'data') and torch.is_tensor(obj.data)):
           if str(obj.device) != 'cpu':
               tensor_sizes.append(obj.size())
               tensors.append(obj)
   except:
       pass
# 大きさ順にsort
size_index = sorted(range(len(tensor_sizes)), key=lambda k: tensor_sizes[k], reverse=True)
# 上位5個のtensorの大きさ
for i in range(0, 5):
   print(tensor_sizes[size_index[i]])
   # print(tensors[size_index[i]])
# tensor 全体の大きさ(bytes)
total_sum = 0
for tensor_size in tensor_sizes:
   current_num = 1
   for num in tensor_size:
       current_num *= num
   total_sum += current_num
print(str(total_sum * 4))