利用Google calab GPU训练模型

在Google calab查看gpu
!/opt/bin/nvidia-smi

#解压
!unzip /content/yolov5-6.2.zip -d /content/yolov5
#进入指定文件目录
%cd /content/yolov5/yolov5-6.2
#加在tensorboard
%load_ext tensorboard
#运行
%tensorboard --logdir==runs/train
#开始训练
!python train.py --rect

在pycharm中,使用组合键ctrl+alt+L进行json格式化。

你可能感兴趣的:(利用Google calab GPU训练模型)