当有多台GPU可用时，如果想指定用某块GPU来训练：
''' CUDA_VISIBLE_DEVICES=gpu_id python main.py '''
本文标题：Pytorch用指定GPU训练
本文链接：https://www.haomeiwen.com/subject/goqfbctx.html
image.png
网友评论