pytorch

利用Google calab GPU训练模型

2022-11-19  本文已影响0人  午字横
在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格式化。

上一篇 下一篇

猜你喜欢

热点阅读