Pytorch

指定GPU训练3中方法

1.指定环境变量，屏蔽第0块gpu

CUDA_VISIBLE_DEVICES = 1 main.py

2.使用os模块

import os
#多块使用逗号隔开
os.environ['CUDA_VISIBLE_DEVICES'] = '1'

3.使用torch.cuda.device

with torch.cuda.device(1):
	...