PyTorch 1.8 リリース

 (pytorch.org)
3 ポイント 投稿者 tkwlsrl 2021-03-14 | 2件のコメント | WhatsAppで共有

2021年3月4日、PyTorch 1.8 がリリースされました。

  • 新しいAPIの追加、分散学習機能のアップデート

  • AMD GPU向けの ROCm もサポート（ベータ）

  • TorchVision、TorchText、TorchAudio のアップデートも提供

  • PyTorch Mobile のチュートリアルを提供（Android / iOS で PyTorch を実行）

関連記事

2件のコメント

 
tkwlsrl 2021-03-14

いつもニュースを楽しく拝見しています。私もひとつ書き込んでみます。

参考までに、PyTorch 1.8 では GPU memory を分割して使える memory fraction 機能が追加されました。

https://pytorch.org/docs/stable/cuda.html

torch.cuda.set_per_process_memory_fraction(fraction, device=None)
 
xguru 2021-03-14

ありがとうございます〜！ 私は機械学習のほうはあまり強くなくて（笑）

こうして整理していただけると助かります。これからも時々お願いします！ ^^