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

PyTorch指定GPU进行训练

3年前 (2022) 程序员胖胖胖虎阿
247 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永久激活工具永久有效)

相关文章

见成效·见未来 | 2022年品牌电商精细化运营论坛开幕在即!
程序员胖胖胖虎阿
232
1 分钟在 Serverless 上部署现代化 Deno Web 应用
程序员胖胖胖虎阿
354
基于java springboot+mybatis爱游旅行平台前台+后台设计实现
程序员胖胖胖虎阿
410
IntelliJ IDEA2022.1破解插件(IDEA全新破解方法永久有效)
程序员胖胖胖虎阿
236
Squoosh - 谷歌出品的免费开源图片压缩工具,图片大小减少90%,支持 API 开发调用
程序员胖胖胖虎阿
1,001
Hugging Face:成为机器学习界的“GitHub”
程序员胖胖胖虎阿
493

暂无评论

再想想
暂无评论...
程序员胖胖胖虎阿 博主
33472文章 15评论 11.5M浏览 0获赞
胖虎的工具箱-编程导航 胖虎的工具箱-编程导航
致力于称为最好的程序员导航网站

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

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