Medical Imaging Interaction Toolkit  2024.12.99-0da743f6
Medical Imaging Interaction Toolkit
QmitknnUNetGPU.h
Go to the documentation of this file.
1 /*============================================================================
2 
3 The Medical Imaging Interaction Toolkit (MITK)
4 
5 Copyright (c) German Cancer Research Center (DKFZ)
6 All rights reserved.
7 
8 Use of this source code is governed by a 3-clause BSD license that can be
9 found in the LICENSE file.
10 
11 ============================================================================*/
12 
13 #ifndef QmitknnUNetGPU_h
14 #define QmitknnUNetGPU_h
15 
16 #include <vector>
17 #include <QString>
19 
25 {
26  QString name;
27  QString memory;
28  unsigned int id;
29 };
30 
36 {
37 private:
38  std::vector<QmitkGPUSpec> m_Gpus;
39 
40 public:
46  ~QmitkGPULoader() = default;
47 
53  int GetGPUCount() const;
54 
60  std::vector<QmitkGPUSpec> GetAllGPUSpecs();
61 };
62 
63 #endif
MITKSEGMENTATIONUI_EXPORT
#define MITKSEGMENTATIONUI_EXPORT
Definition: MitkSegmentationUIExports.h:15
QmitkGPULoader
Class to load and save GPU information for further validation.
Definition: QmitknnUNetGPU.h:35
QmitkGPUSpec::name
QString name
Definition: QmitknnUNetGPU.h:26
QmitkGPUSpec::id
unsigned int id
Definition: QmitknnUNetGPU.h:28
MitkSegmentationUIExports.h
QmitkGPUSpec::memory
QString memory
Definition: QmitknnUNetGPU.h:27
QmitkGPUSpec
Struct to store GPU info.
Definition: QmitknnUNetGPU.h:24