Skip to content

Commit 8efa1e7

Browse files
committed
ref: do not import torch from dcnum, actually has no effect
1 parent d274be3 commit 8efa1e7

File tree

2 files changed

+4
-2
lines changed

2 files changed

+4
-2
lines changed

CHANGELOG

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,5 @@
1+
0.9.13
2+
- ref: do not import torch from dcnum, actually has no effect
13
0.9.12
24
- fix: import torch from dcnum to make sure variables are set correctly
35
0.9.11

chipstream/gui/main_window.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,7 @@
1515
import psutil
1616
from PyQt6 import uic, QtCore, QtWidgets
1717
from 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

2020
from ..path_cache import PathCache
2121
from .._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 = []

0 commit comments

Comments
 (0)