俵言

しがない社会人が書く、勉強とかのこと。最近は機械学習や kaggle 関連がメイン。

pytorch

poetry で直接指定する wheel file を url ではなく local path にすることで依存関係解決の度に download されるのを防げた件

背景 前回の記事では、poetry でライブラリの wheel file を直接 URL で指定することで依存関係の解決が長すぎる 問題*1を解消し、同一の poetry 環境に PyTorch と RAPIDs を共存させました。 tawara.hatenablog.com これ自体は大成功だったのですが、wheel…

CUDA 更新のついでに PyTorch と RAPIDS を poetry の同一環境にゴリ押しで共存させた件

はじめに 皆さんお久しぶりです。本当にいつぶりかわからないくらいですが、久々に CUDA の version を更新をするにあたって書きたいネタが出来たのでまとめることにしました。 更新のたびにやっている作業や関連リンクをまとめるというのが当初の目的でした…

小ネタ:Pytorch で Automatic Mixed Precision (AMP) の ON/OFF をするときの話

Pytorch の AMP 使うために if 文で分岐してたけど実は要らなかったという話。