QmitknnUNetGPU.h

Go to the documentation of this file.

/*============================================================================
 
The Medical Imaging Interaction Toolkit (MITK)
 
Copyright (c) German Cancer Research Center (DKFZ)
All rights reserved.
 
Use of this source code is governed by a 3-clause BSD license that can be
found in the LICENSE file.
 
============================================================================*/
 
#ifndef QmitknnUNetGPU_h
#define QmitknnUNetGPU_h
 
#include <vector>
#include <QString>
#include <MitkSegmentationUIExports.h>
 
struct QmitkGPUSpec
{
  QString name;
  QString memory;
  unsigned int id;
};
 
class MITKSEGMENTATIONUI_EXPORT QmitkGPULoader
{
private:
  std::vector<QmitkGPUSpec> m_Gpus;
 
public:
  QmitkGPULoader();
  ~QmitkGPULoader() = default;
 
  int GetGPUCount() const;
 
  std::vector<QmitkGPUSpec> GetAllGPUSpecs();
};
 
#endif