Breaking News: Grepper is joining You.com. Read the official announcement!

colab not training always giving cuda out of memory error eventhough memory is available

Enthusiastic Earthworm answered on May 15, 2021 Popularity 1/10 Helpfulness 1/10

answer colab not training always giving cuda out of memory error eventhough memory is available

colab not training always giving cuda out of memory error eventhough memory is available

Comment

# memory footprint support libraries/code
!ln -sf /opt/bin/nvidia-smi /usr/bin/nvidia-smi
!pip install gputil
!pip install psutil
!pip install humanize

import psutil
import humanize
import os
import GPUtil as GPU

GPUs = GPU.getGPUs()
# XXX: only one GPU on Colab and isn’t guaranteed
gpu = GPUs[0]
def printm():
    process = psutil.Process(os.getpid())
    print("Gen RAM Free: " + humanize.naturalsize(psutil.virtual_memory().available), " |     Proc size: " + humanize.naturalsize(process.memory_info().rss))
    print("GPU RAM Free: {0:.0f}MB | Used: {1:.0f}MB | Util {2:3.0f}% | Total     {3:.0f}MB".format(gpu.memoryFree, gpu.memoryUsed, gpu.memoryUtil*100, gpu.memoryTotal))
printm()

Popularity 1/10 Helpfulness 1/10 Language python

Source: stackoverflow.com

Tags: cuda memory out-of-memory python training

Link to this answer
Share Copy Link

Contributed on May 15 2021

Enthusiastic Earthworm

0 Answers Avg Quality 2/10

colab not training always giving cuda out of memory error eventhough memory is available

Contents

More Related Answers

colab not training always giving cuda out of memory error eventhough memory is available

Grepper

Documentation

Social

Legal

Contact

Oops, You will need to install Grepper and log-in to perform this action.