登录社区云,与社区用户共同成长
邀请您加入社区
$ nvidia-smi
查看当前节点下有几块GPU,以及每一块的使用情况
在运行程序时指定使用哪块GPU
$ CUDA_VISIBLE_DEVICES=5 python main.py --cuda
欢迎来到FlagOS开发社区,这里是一个汇聚了AI开发者、数据科学家、机器学习爱好者以及业界专家的活力平台。我们致力于成为业内领先的Triton技术交流与应用分享的殿堂,为推动人工智能技术的普及与深化应用贡献力量。
更多推荐
AI Infra 正悄悄融入传统 OS 生态
智算前沿:下一代架构及基础软件
扫一扫分享内容
为遵守国家网络实名制规定,未绑定将限制内容发布与互动
所有评论(0)