pytorch多GPU并行运算

可以快速上手的开发文档

#pytorch多GPU并行运算

pytorch多GPU并行运算的实现

Pytorch多GPU运行设置可用GPU环境变量。例如,使用0号和1号GPU'os.environ["CUDA_VISIBLE_DEVICES"]='0,1'..