http://zh.d2l.ai/chapter_computational-performance/multiple-gpus-concise.html
注意，这部分实例代码的原始库进行了相应的修改，请使用英文版中的代码。