Pytorch用指定GPU训练

image.png

当有多台GPU可用时,如果想指定用某块GPU来训练:

'''
CUDA_VISIBLE_DEVICES=gpu_id python main.py
'''

你可能感兴趣的:(Pytorch用指定GPU训练)