PyTorch指定GPU进行训练

lewis 6年前 (2019-06-21) 阅读数 9 #技术

原文链接

1 直接在终端中设定：

CUDA_VISIBLE_DEVICES=1 python main.py

2 python代码中设定：

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

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

代码的路

本文仅代表作者观点，不代表博信信息网立场。