美文网首页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