美文网首页pytorch
利用Google calab GPU训练模型

利用Google calab GPU训练模型

作者: 午字横 | 来源:发表于2022-11-19 00:22 被阅读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格式化。

相关文章

网友评论

    本文标题:利用Google calab GPU训练模型

    本文链接:https://www.haomeiwen.com/subject/jlohxdtx.html