代码的路

V2

2022/08/18阅读:12主题:默认主题

PyTorch指定GPU进行训练

原文链接

1直接终端中设定:

CUDA_VISIBLE_DEVICES=1 python main.py

2 python代码中设定:

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

   

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

代码的路

分类:

后端

标签:

Python

作者介绍

代码的路
V2

公众号:代码的路