胖虎的工具箱-编程导航 胖虎的工具箱-编程导航 胖虎的工具箱-编程导航 胖虎的工具箱-编程导航
  • 站点推荐
    • Blog
    • 开发必备
      • 学习平台
      • 技术社区
      • 练习专区
      • 编程工具
    • 排行榜
    • 文档手册
    • 在线工具
    • 协同工具
    • 网站提交
  • 开发必备
    • 学习平台
    • 技术社区
    • 练习专区
    • 编程工具
  • 大厂博客
  • 摸鱼专区
  • 协同工具
  • 文档手册
  • 在线工具
  • 招聘求职
  • 简历模板
  • 在线音乐
  • 开发软件
  • 工具软件
  • Win系统工具
    • 效率工具
  • 网盘搜索
  • Mac 软件
  • 图片素材
  • 图库壁纸
  • 站长工具
  • 自媒体
    • 分析工具
    • 文案
    • 视频无水印下载
    • 字幕处理
    • 工具
    • 违禁词检测
    • 音频处理
  • 网站提交
胖虎的工具箱-编程导航 胖虎的工具箱-编程导航
  • Blog
  • 开发必备
    • 学习平台
    • 技术社区
    • 练习专区
    • 编程工具
  • 排行榜
  • 文档手册
  • 在线工具
  • 协同工具
  • 网站提交
首页•未分类•PyTorch指定GPU进行训练

PyTorch指定GPU进行训练

3年前 (2022) 程序员胖胖胖虎阿
283 0 0

原文链接

1 直接在终端中设定:

CUDA_VISIBLE_DEVICES=1 python main.py

2 python代码中设定:

import os
os.environ["CUDA_VISIBLE_DEVICES"] = "0,1"

 
 

学习更多编程知识,请关注我的公众号:

代码的路

PyTorch指定GPU进行训练

版权声明:程序员胖胖胖虎阿 发表于 2022年11月3日 下午12:16。
转载请注明:PyTorch指定GPU进行训练 | 胖虎的工具箱-编程导航
上一篇

Intellij IDEA永久激活破解教程(IntelliJ IDEA2022.2激活码,绝对有效)

下一篇

IntelliJ IDEA2022.1免费激活码(windows/mac idea永久激活工具永久有效)

相关文章

IDEA安装及注册码(windows/mac idea全新破解教程,最新有效IDEA激活码)
程序员胖胖胖虎阿
259
GET和POST的区别
程序员胖胖胖虎阿
296
Selinux
程序员胖胖胖虎阿
315
后端中使用分页的几种方法(建议收藏)
程序员胖胖胖虎阿
317
IntelliJ IDEA2022破解插件(IntelliJ IDEA2019破解插件成功激活)
程序员胖胖胖虎阿
269
IntelliJ IDEA2022.2破解方法(IDEA全新破解教程,永久有效)
程序员胖胖胖虎阿
234

暂无评论

再想想
暂无评论...

JetBrains全家桶正版账号

官方正版 JetBrains 账号

登录即可使用全家桶 不限操作系统 官方授权 支持全家桶所有软件 不限版本
🔥 最低 32元/年
胖虎的工具箱-编程导航 胖虎的工具箱-编程导航
致力于称为最好的程序员导航网站

友链申请 免责声明 关于我们

Copyright © 2022  胖虎的工具箱-编程导航   陇ICP备2022001249号  
常用
搜索
工具
社区
生活
求职
    热门推荐: