Код Pytorch испортит мою оперативную память при использовании torch.zeros () ⇐ Python

[Anonymous]

У меня есть функция для измерения выделенной оперативной памяти Python в мегабайтах:

Код: Выделить всё

def getram(): print(psutil.Process(os.getpid()).memory_info().rss / 1024**2)< /code> < /p>
, а также у меня есть:
device = "cuda"

Моя проблема в том, что следующий код выделяет ОЗУ, и я схожу с ума из -за этого. Есть ли у него реальное решение или я должен принять свою судьбу и переключиться на C ++ или что -то в этом роде? < /P>
Код: < /p>

Код: Выделить всё

getram()

def load_dataset(dir, filenames):
dataset = torch.zeros((len(filenames),3,256,256), device=device)
getram()
for i, filename in enumerate(filenames):
f = read_image(f"{dir}/{filename}")
if 3 != f.shape[0]: print(filename)
dataset[i] = f.to(device)
getram()
return dataset

dataset = load_dataset(dataset_dir, dataset_filenames)

getram()
< /code>
Код распечатал следующее: < /p>
533.28125
661.2890625
678.27734375
678.27734375

Как вы можете видеть, как только я создаю пустой тенсор с Torch.zeros () , он занимает оперативную память без причины.>

Подробнее здесь: https://stackoverflow.com/questions/795 ... orch-zeros