Pytorch用指定GPU训练
2019-11-01 本文已影响0人
啦啦啦_e62f
image.png
当有多台GPU可用时，如果想指定用某块GPU来训练：
'''
CUDA_VISIBLE_DEVICES=gpu_id python main.py
'''
image.png
image.png
image.png
当有多台GPU可用时，如果想指定用某块GPU来训练：
'''
CUDA_VISIBLE_DEVICES=gpu_id python main.py
'''
