PyTorch指定GPU进行训练

举报
代码的路 发表于 2023/01/13 10:08:02 2023/01/13
【摘要】 原文链接1 直接在终端中设定:CUDA_VISIBLE_DEVICES=1 python main.py2 python 代码中设定:import osos.environ["CUDA_VISIBLE_DEVICES"] = "0,1"  学习更多编程知识,请关注我的公众号:代码的路

原文链接

1 直接在终端中设定:

CUDA_VISIBLE_DEVICES=1 python main.py

2 python 代码中设定:

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

 
 

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

代码的路

【版权声明】本文为华为云社区用户原创内容,转载时必须标注文章的来源(华为云社区)、文章链接、文章作者等基本信息, 否则作者和本社区有权追究责任。如果您发现本社区中有涉嫌抄袭的内容,欢迎发送邮件进行举报,并提供相关证据,一经查实,本社区将立刻删除涉嫌侵权内容,举报邮箱: cloudbbs@huaweicloud.com
  • 点赞
  • 收藏
  • 关注作者

评论(0

0/1000
抱歉,系统识别当前为高风险访问,暂不支持该操作

全部回复

上滑加载中

设置昵称

在此一键设置昵称,即可参与社区互动!

*长度不超过10个汉字或20个英文字符,设置后3个月内不可修改。

*长度不超过10个汉字或20个英文字符,设置后3个月内不可修改。