Glenn Jocherandpre-commit-ci[bot]
7d56d45124
* Update notebook * Update notebook * update string * update string * Updates * Updates * Updates * check both ipython and psutil * remove sample_data if is_colab * cleanup * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
38 lines
1.1 KiB
Python
38 lines
1.1 KiB
Python
# YOLOv5 🚀 by Ultralytics, GPL-3.0 license
|
|
"""
|
|
utils/initialization
|
|
"""
|
|
|
|
|
|
def notebook_init(verbose=True):
|
|
# Check system software and hardware
|
|
print('Checking setup...')
|
|
|
|
import os
|
|
import shutil
|
|
|
|
from utils.general import check_requirements, emojis, is_colab
|
|
from utils.torch_utils import select_device # imports
|
|
|
|
check_requirements(('psutil', 'IPython'))
|
|
import psutil
|
|
from IPython import display # to display images and clear console output
|
|
|
|
if is_colab():
|
|
shutil.rmtree('sample_data', ignore_errors=True) # remove colab /sample_data directory
|
|
|
|
if verbose:
|
|
# System info
|
|
# gb = 1 / 1000 ** 3 # bytes to GB
|
|
gib = 1 / 1024 ** 3 # bytes to GiB
|
|
ram = psutil.virtual_memory().total
|
|
total, used, free = shutil.disk_usage("/")
|
|
display.clear_output()
|
|
s = f'({os.cpu_count()} CPUs, {ram * gib:.1f} GB RAM, {(total - free) * gib:.1f}/{total * gib:.1f} GB disk)'
|
|
else:
|
|
s = ''
|
|
|
|
select_device(newline=False)
|
|
print(emojis(f'Setup complete ✅ {s}'))
|
|
return display
|