AllenCell / AllenCell/allencell-segmenter-ml

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

未关闭
#491 0 条评论 0 个 reaction 已指派 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 摘要。