您当前的位置:首页 > 计算机 > 编程开发 > 人工智能

mmdetection训练 显卡选择

时间:01-16来源:作者:点击数:

训练细节

在终端执行.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开始)

方便获取更多学习、工作、生活信息请关注本站微信公众号城东书院 微信服务号城东书院 微信订阅号
推荐内容
相关内容
栏目更新
栏目热门