训练细节

在终端执行.py文件时通过CUDA_VISIBLE_DEVICES来指定:

在终端运行python程序的前面指定,比如:

CUDA_VISIBLE_DEVICES=1 python train.py

实验效果如下:

在终端运行脚本或者jupyter notebook下运行:

前面加上CUDA_VISIBLE_DEVICES=1
训练代码

实时查看显卡性能

watch -n 1 -d nvidia-smi

显卡实时显示

就会实时刷新了,可以查看显卡的用量等,哪块卡是空闲的(0开始)

GitHub 加速计划 / mm / mmdetection
28.78 K
9.33 K
下载
open-mmlab/mmdetection: 是一个基于 PyTorch 的人工智能物体检测库,支持多种物体检测算法和工具。该项目提供了一个简单易用的人工智能物体检测库,可以方便地实现物体的检测和识别,同时支持多种物体检测算法和工具。
最近提交(Master分支:1 个月前 )
cfd5d3a9 8 个月前
2390ebc3 8 个月前
Logo

旨在为数千万中国开发者提供一个无缝且高效的云端环境,以支持学习、使用和贡献开源项目。

更多推荐