mmdetection训练显卡选择

滴滴滴'cv

9556人浏览 · 2020-06-17 10:30:05

滴滴滴'cv · 2020-06-17 10:30:05 发布

选择显卡块数

训练细节
实时查看显卡性能

训练细节

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

AtomGit开源社区

AtomGit 是由开放原子开源基金会联合 CSDN 等生态伙伴共同推出的新一代开源与人工智能协作平台。平台坚持“开放、中立、公益”的理念，把代码托管、模型共享、数据集托管、智能体开发体验和算力服务整合在一起，为开发者提供从开发、训练到部署的一站式体验。

更多推荐

1.8B 体积、33 种语言互译｜腾讯混元 HY-MT1.5-1.8B 多语言机器翻译模型上线

在跨语言交流日益频繁的今天，阅读外语菜单、处理多语言邮件、与不同语言背景的人沟通，已经成为很多人日常工作与生活的一部分。过去，这类需求往往依赖联网翻译工具，而如今，—— 一部设备即可支持的相互翻译。当 AI 不再只是“逐字直译”，而是开始理解语境、风格与语言之间的细微差异，机器翻译就真正具备了今天为大家介绍一款高质量、多语言、支持端侧部署的机器翻译模型 ——，现已上线 AtomGit AI 社区，