PyTorch 1.7發佈,支持CUDA 11、Windows分佈式訓練

機器之心報道 參與:魔王、小舟 昨日,PyTorch 團隊發佈 PyTorch 1.7 版本。該版本增添了很多新特性,如支持 CUDA 11、Windows 分佈式訓練、增加了支持快速傅里葉變換(FFT)的新型 API 等。 PyTorch 1.7 版本包含很多新的 API,如支持 NumPy 兼容的 FFT 操作、性能分析工具,以及對基於分佈式數據並行(DDP)和基於遠程過程調用(RPC)的分佈
相關文章
相關標籤/搜索