File tree Expand file tree Collapse file tree 2 files changed +4
-2
lines changed
Expand file tree Collapse file tree 2 files changed +4
-2
lines changed Original file line number Diff line number Diff line change 1+ 0.9.13
2+ - ref: do not import torch from dcnum, actually has no effect
130.9.12
24 - fix: import torch from dcnum to make sure variables are set correctly
350.9.11
Original file line number Diff line number Diff line change 1515import psutil
1616from PyQt6 import uic , QtCore , QtWidgets
1717from PyQt6 .QtCore import QStandardPaths
18- from dcnum . segm . segm_torch . torch_setup import torch
18+ from torch . cuda import is_available as cuda_is_available
1919
2020from ..path_cache import PathCache
2121from .._version import version
@@ -71,7 +71,7 @@ def __init__(self, *arguments):
7171 self .comboBox_segmenter .addItem ("Thresholding" , "thresh" )
7272 # torch
7373 self .comboBox_segmenter .addItem ("Machine-learning model" , "torch" )
74- use_gpu = torch . cuda . is_available ()
74+ use_gpu = cuda_is_available ()
7575 self .checkBox_torch_use_gpu .setVisible (use_gpu )
7676 self .checkBox_torch_use_gpu .setChecked (use_gpu )
7777 available_models = []
You can’t perform that action at this time.
0 commit comments