1.服务器(Linux终端服务器)Bourne Again Shell
(1).重新加载当前用户的 Bash 配置文件,让最近做的修改立即生效,而不用关闭再重新打开终端。
source ~/.bashrc
(2).验证Linux 版本的 PyTorch 安装包,,是否ok
python -c "
import torch
print('PyTorch version:', torch.__version__)
print('CUDA available:', torch.cuda.is_available())
print('GPU count:', torch.cuda.device_count())
"
输出结果
PyTorch version: 2.4.1+cu118
CUDA available: True
GPU count: 2
(3).进入项目目录
cd ~/project/yolov8/ultralytics-main
(4).执行代码,开始训练
nohup python c_train_guo_fuwuqi.py > train.log 2>&1 &
echo $! > train.pid
(5).查看日志
tail -f train.log
