Pytorch 单机多GPU运行
一、单机单GPU1、set current device (gpu id)# the first methordCUDA_VISIBLE_DEVICES=gpi_id python XXX.py# the second methordtorch.cuda.set_device(gpu_id)2、put the object(tensor,variable,model…) to G...
原创
2019-05-23 22:35:56 ·
2419 阅读 ·
0 评论