Changed: Parse nvidia-smi output to determine video memory
This commit is contained in:
parent
f793077ac0
commit
51dba01e7d
1 changed files with 146 additions and 67 deletions
|
@ -601,6 +601,83 @@ sint CSystemUtils::getTotalVideoMemory()
|
||||||
#elif defined(NL_OS_MAC)
|
#elif defined(NL_OS_MAC)
|
||||||
// the right method is using OpenGL
|
// the right method is using OpenGL
|
||||||
#else
|
#else
|
||||||
|
if (res == -1)
|
||||||
|
{
|
||||||
|
// use nvidia-smi
|
||||||
|
std::string command = "nvidia-smi -q -d MEMORY";
|
||||||
|
|
||||||
|
std::string out = getCommandOutput(command);
|
||||||
|
|
||||||
|
if (out.empty())
|
||||||
|
{
|
||||||
|
nlwarning("Unable to launch %s", command.c_str());
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
std::vector<std::string> lines;
|
||||||
|
explode(out, std::string("\n"), lines, true);
|
||||||
|
|
||||||
|
// process each line
|
||||||
|
for(uint i = 0; i < lines.size(); ++i)
|
||||||
|
{
|
||||||
|
// Total : 62 MB
|
||||||
|
|
||||||
|
std::string line = lines[i];
|
||||||
|
|
||||||
|
// find Total line
|
||||||
|
std::string::size_type pos = line.find("Total");
|
||||||
|
if (pos == std::string::npos) continue;
|
||||||
|
pos += 6;
|
||||||
|
|
||||||
|
// find separator
|
||||||
|
pos = line.find(':', pos);
|
||||||
|
if (pos == std::string::npos) continue;
|
||||||
|
pos += 2;
|
||||||
|
|
||||||
|
// find units
|
||||||
|
std::string::size_type posUnits = line.find(' ', pos);
|
||||||
|
if (posUnits == std::string::npos) continue;
|
||||||
|
++posUnits;
|
||||||
|
|
||||||
|
// found device ID
|
||||||
|
std::string memory = line.substr(pos, posUnits-pos-1);
|
||||||
|
std::string units = line.substr(posUnits);
|
||||||
|
|
||||||
|
// convert video memory to sint
|
||||||
|
if (NLMISC::fromString(memory, res))
|
||||||
|
{
|
||||||
|
if (units == "MB")
|
||||||
|
{
|
||||||
|
res *= 1024;
|
||||||
|
}
|
||||||
|
else if (units == "GB")
|
||||||
|
{
|
||||||
|
res *= 1024 * 1024;
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
// reset to use other methods
|
||||||
|
res = -1;
|
||||||
|
|
||||||
|
nlwarning("nvidia-smi reported %d %s as wrong video memory units", res, units.c_str());
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
|
||||||
|
nlinfo("nvidia-smi reported %d KiB of video memory", res);
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
// reset to use other methods
|
||||||
|
res = -1;
|
||||||
|
}
|
||||||
|
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if (res == -1)
|
||||||
|
{
|
||||||
// under Linux, no method is really reliable...
|
// under Linux, no method is really reliable...
|
||||||
NLMISC::CIFile file;
|
NLMISC::CIFile file;
|
||||||
|
|
||||||
|
@ -625,6 +702,7 @@ sint CSystemUtils::getTotalVideoMemory()
|
||||||
if (pos != std::string::npos)
|
if (pos != std::string::npos)
|
||||||
{
|
{
|
||||||
// [ 20.883] (--) NVIDIA(0): Memory: 2097152 kBytes
|
// [ 20.883] (--) NVIDIA(0): Memory: 2097152 kBytes
|
||||||
|
// [ 28.515] (--) NVIDIA(0): Memory: 262144 kBytes
|
||||||
pos = line.find("Memory: ", pos);
|
pos = line.find("Memory: ", pos);
|
||||||
|
|
||||||
// found memory line
|
// found memory line
|
||||||
|
@ -669,11 +747,12 @@ sint CSystemUtils::getTotalVideoMemory()
|
||||||
break;
|
break;
|
||||||
}
|
}
|
||||||
|
|
||||||
// TODO: other drivers: nv, fglrx (ATI), radeon (ATI)
|
// TODO: other drivers: fglrx (ATI), radeon (ATI)
|
||||||
}
|
}
|
||||||
|
|
||||||
file.close();
|
file.close();
|
||||||
}
|
}
|
||||||
|
}
|
||||||
|
|
||||||
if (res == -1)
|
if (res == -1)
|
||||||
{
|
{
|
||||||
|
|
Loading…
Reference in a new issue