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"

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

代码的路

0/1000

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

上滑加载中

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

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