added nvidia gpu speed control

Author: natankeddem

hush/hardware/factory.py

@@ -71,6 +71,8 @@ async def driver(
                     cls.drivers[host][group]["instance"] = cisco.M5(host)
                 elif name == "OpenJBOD":
                     cls.drivers[host][group]["instance"] = openjbod.Rp2040(host)
+                elif name == "Nvidia":
+                    cls.drivers[host][group]["instance"] = nvidia.Gpu(host)
                 else:
                     cls.drivers[host][group]["instance"] = None
             if group == "cpu":

hush/hardware/nvidia.py

@@ -22,3 +22,9 @@ async def get_temp(self):
             logger.info(f"{self.hostname} failed to get gpu temperature:")
             logger.info(f"result = {result}")
             raise e
+
+    async def set_speed(self, speed):
+        await self.ssh.shell(f'export DISPLAY=:0 && nvidia-settings -c $DISPLAY  -a "GPUFanControlState=1" -a "GPUTargetFanSpeed={speed}"')
+
+    async def close(self):
+        await self.ssh.shell('export DISPLAY=:0 && nvidia-settings -c $DISPLAY  -a "GPUFanControlState=0"')

hush/tabs/configure.py

@@ -45,6 +45,7 @@
     "Cisco M4",
     "Cisco M5",
     "OpenJBOD",
+    "Nvidia",
 ]
 drive_sensor_names = ["None", "SMART All", "SMART Discrete"]
 gpu_sensor_names = ["None", "Nvidia", "Supermicro"]