AllenCell / AllenCell/allencell-segmenter-ml

Check for compatible cuda and pytorch version and show dialog box with instructions if needed

オープン
#491 コメント 0 件 リアクション 0 件 担当者 1 名 @hughes036 に割り当て済み GitHub で見る
🚨 high priority🚨
主要言語
Python
スター
5
フォーク
3
PR マージ指標
30日以内にマージされた PR はありません

説明

When the plugin starts up we should use `torch.cuda.available()` to check if the user's cuda driver is compatible with the version of pytorch installed on their system.

if incompatible, do `nvidia-smi` to check if they have a gpu on their system, and if they do,`nvcc --version` to check the version of cuda currently on their system, which may require error handling if the user has no GPU available on their system. Display all relevant information about their gpu/cuda drivers, a link to pytorch versions and a link to our documentation on how to install pytorch.

show this message one time, and set an application property (like experiments home) to not display this message again.

include the option to display this in our dropdown menu

コントリビューションガイド

このリポジトリのコントリビューションガイドは索引されていません

評価

この issue はまだ評価されていません。

新しい issue をメールで受け取る

初心者向けの GitHub issue を短くまとめたダイジェスト。