当前位置: 首页 > article >正文

【学习】电脑上有多个GPU,命令行指定GPU进行训练。

使用如下指令可以指定使用的GPU。

CUDA_VISIBLE_DEVICES=1 

假设要使用第二个GPU进行训练。

CUDA_VISIBLE_DEVICES=1 python train.py


http://www.kler.cn/a/322998.html

相关文章:

  • 2024山西省网络建设运维第十八届职业院校技能大赛解析答案(5. nginx 和 tomcat 服务)
  • 逆向攻防世界CTF系列37-crackme
  • Linux驱动开发第2步_“物理内存”和“虚拟内存”的映射
  • xcode-select: error: tool ‘xcodebuild‘ requires Xcode, but active developer
  • 大数据实验9:Spark安装和编程实践
  • 一文3000字从0到1带你进行Mock测试(建议收藏)
  • C语言习题~day33
  • 【Unity保龄球项目】的实现逻辑以及代码解释
  • Python Daphne库:ASGI服务的高效Web服务器
  • 使用FFmpeg压缩MP3格式音频
  • 利用模糊综合评价法进行数值评分计算——代码实现
  • 基于Java开发的(控制台)模拟的多用户多级目录的文件系统
  • Redis的主要特点及运用场景
  • 【Linux】ubuntu 16.04 搭建jdk 11 环境(亲测可用)
  • 数据结构:特殊矩阵 及其存储
  • 策略路由控制选路
  • apt update时出现证书相关问题,可以关闭apt验证
  • 【Redis 源码】3dict字典数据结构
  • 打点 - 泛微 E-Cology WorkflowServiceXml
  • FPGA学习(3)-38译码器实现
  • LLM基础概念:Prompt
  • LeetCode --- 416周赛
  • Unity图形用户界面!*★,°*:.☆( ̄▽ ̄)/$:*.°★* 。(万字解析)
  • 常用性能优化方法
  • jdk tomcat 镜像制作
  • Activiti7《第九式:破气式》——流畅驱动工作流进程。面试题大全