Bitcoin Forum

Traceback (most recent call last):
  File "./gpu-watch.py", line 127, in <module>
    w.watch()
  File "./gpu-watch.py", line 75, in watch
    self.watch_card(card)
  File "./gpu-watch.py", line 79, in watch_card
    temp = card.temp()
  File "./gpu-watch.py", line 21, in temp
    line = subprocess.check_output(
AttributeError: 'module' object has no attribute 'check_output'

$ python -V
Python 2.6.6

#!/usr/bin/python
# -*- coding: utf-8 -*-

import sys
import time
import re
import os
import subprocess
from string import split

settings = {
	'poll_time': 60,
	'temp_low': 70.0,
        'temp_high': 76.0,
	'aticonfig_path': '/usr/bin/aticonfig'
}

class GPU():
	def __init__(self, id, cardid, description):
		self.id = id
		self.cardid = cardid
		self.description = description

	def temp(self):
		os.environ['DISPLAY'] = ":0"
		try:
			line = subprocess.check_output(
				[settings['aticonfig_path'],
				 "--adapter=" + str(self.id),
				 "--odgt"])
			m = re.search('Temperature - (\d+\.\d+)\s*C', line)
			if m is None:
				return None

			return float(m.group(1))
		except OSError:
			return None

class Card():
	def __init__(self, id, gpus):
		self.id = id
		self.x11_id = ':0.' + str(id)
		self.gpus = gpus

	def fanspeed(self, val=None):
		os.environ['DISPLAY'] = self.x11_id
		if val is not None:
			subprocess.call( [settings['aticonfig_path'],
				 "--pplib-cmd", "set fanspeed 0 " + str(val)])
		else:
			line = subprocess.check_output(
				[settings['aticonfig_path'],
				 "--pplib-cmd", "get fanspeed 0"])
			m = re.search('Fan Speed: (\d+)%', line)
			if m is None:
				return False
			return int(m.group(1))

class Watcher():
	def __init__(self):
		self.cards = []
		os.environ['DISPLAY'] = ":0"
		out = subprocess.check_output([settings['aticonfig_path'],
			"--list-adapters"])
		card = 0
		gpu = 0
		lines = split(out, "\n")
		for line in lines:
			r = re.search('^[ \*]+(\d+)\. [\d:\.]+ (.+)$', line)
			if r is None:
				continue
			gpuid = int(r.group(1))
			desc = r.group(2)
			if gpu % 2 == 0:
				self.cards.append(Card(card, []))
			self.cards[-1].gpus.append(GPU(gpuid, card, desc))
			print "gpu %d card %d desc %s" % (gpuid, card, desc)
			if gpu % 2 == 1:
				card = card + 1
			gpu = gpu + 1

	def watch(self):
		while True:
			for card in self.cards:
				if card is None:
					continue
				self.watch_card(card)
			time.sleep(settings['poll_time'])
			print "----------------------------------"

	def watch_card(self, card):
		fanspeed = card.fanspeed()
		fandelta = 0
		print "Card %d: fan speed %d%%" % (card.id, fanspeed)
		for gpu in card.gpus:
			temp = gpu.temp()
			print "Card %d GPU %d: Temperature %.2f °C" % (
				card.id, gpu.id, temp)
			if temp >= 90.0:
				fandelta = 100 - fanspeed
				break
			if temp > settings['temp_high']:
				if fanspeed <= 90:
					fandelta = +10
				else:
					fandelta = 100 - fanspeed
				break
			elif temp < settings['temp_low']:
				if fanspeed <= 5:
					continue
				fandelta = -5
		if fandelta != 0 and fanspeed + fandelta >= 10:
			print "Card %d: Adjusting fan %d%% to %d%%" % (
				card.id, fandelta, fanspeed + fandelta)
			card.fanspeed(fanspeed + fandelta)

if __name__ == '__main__':
	if len(sys.argv) > 1:
		try:
			f = open(sys.argv[1])
			for line in f:
				m = re.search('^(\w+)\s*=\s*(\S.*)$', line)
				if m is None:
					continue
				settings[m.group(1)] = m.group(2)
			f.close()
		except (OSError, IOError), e:
			pass

	settings['poll_time'] = int(settings['poll_time'])
	settings['temp_low'] = float(settings['temp_low'])
	settings['temp_high'] = float(settings['temp_high'])
	print repr(settings)
	w = Watcher()
	w.watch()

#The recipe gives simple implementation of a Discrete Proportional-Integral-Derivative (PID) controller. PID controller gives output value for error between desired reference input and measurement feedback to minimize error value.
#More information: http://en.wikipedia.org/wiki/PID_controller
#
#cnr437@gmail.com
#
#######	Example	#########
#
#p=PID(3.0,0.4,1.2)
#p.setPoint(5.0)
#while True:
#     pid = p.update(measurement_value)
#
#


class PID:
	"""
	Discrete PID control
	"""

	def __init__(self, P=2.0, I=0.0, D=1.0, Derivator=0, Integrator=0, Integrator_max=500, Integrator_min=-500):

		self.Kp=P
		self.Ki=I
		self.Kd=D
		self.Derivator=Derivator
		self.Integrator=Integrator
		self.Integrator_max=Integrator_max
		self.Integrator_min=Integrator_min

		self.set_point=0.0
		self.error=0.0

	def update(self,current_value):
		"""
		Calculate PID output value for given reference input and feedback
		"""

		self.error = self.set_point - current_value

		self.P_value = self.Kp * self.error
		self.D_value = self.Kd * ( self.error - self.Derivator)
		self.Derivator = self.error

		self.Integrator = self.Integrator + self.error

		if self.Integrator > self.Integrator_max:
			self.Integrator = self.Integrator_max
		elif self.Integrator < self.Integrator_min:
			self.Integrator = self.Integrator_min

		self.I_value = self.Integrator * self.Ki

		PID = self.P_value + self.I_value + self.D_value

		return PID

	def setPoint(self,set_point):
		"""
		Initilize the setpoint of PID
		"""
		self.set_point = set_point
		self.Integrator=0
		self.Derivator=0

	def setIntegrator(self, Integrator):
		self.Integrator = Integrator

	def setDerivator(self, Derivator):
		self.Derivator = Derivator

	def setKp(self,P):
		self.Kp=P

	def setKi(self,I):
		self.Ki=I

	def setKd(self,D):
		self.Kd=D

	def getPoint(self):
		return self.set_point

	def getError(self):
		return self.error

	def getIntegrator(self):
		return self.Integrator

	def getDerivator(self):
		return self.Derivator

#!/usr/bin/python
# -*- coding: utf-8 -*-

import sys
import time
import re
import os
import subprocess
import psutil
import signal
from string import split

settings = {
	'poll_time': 60,
	'temp_low': 70.0,
        'temp_high': 76.0,
	'temp_fan_100': 90.0,
	'temp_cooldown': 93.0,
	'cooldown_period': 10,
	'aticonfig_path': '/usr/bin/aticonfig'
}

class GPU():
	def __init__(self, id, cardid, description):
		self.id = id
		self.cardid = cardid
		self.description = description

	def temp(self):
		os.environ['DISPLAY'] = ":0"
		try:
			line = subprocess.check_output(
				[settings['aticonfig_path'],
				 "--adapter=" + str(self.id),
				 "--odgt"])
			m = re.search('Temperature - (\d+\.\d+)\s*C', line)
			if m is None:
				return None

			return float(m.group(1))
		except OSError:
			return None

class Card():
	def __init__(self, id, gpus):
		self.id = id
		self.x11_id = ':0.' + str(id)
		self.gpus = gpus

	def fanspeed(self, val=None):
		os.environ['DISPLAY'] = self.x11_id
		if val is not None:
			subprocess.call( [settings['aticonfig_path'],
				 "--pplib-cmd", "set fanspeed 0 " + str(val)])
		else:
			line = subprocess.check_output(
				[settings['aticonfig_path'],
				 "--pplib-cmd", "get fanspeed 0"])
			m = re.search('Fan Speed: (\d+)%', line)
			if m is None:
				return False
			return int(m.group(1))

class Watcher():
	def __init__(self):
		self.cards = []
		os.environ['DISPLAY'] = ":0"
		out = subprocess.check_output([settings['aticonfig_path'],
			"--list-adapters"])
		card = 0
		gpu = 0
		lines = split(out, "\n")
		for line in lines:
			r = re.search('^[ \*]+(\d+)\. [\d:\.]+ (.+)$', line)
			if r is None:
				continue
			gpuid = int(r.group(1))
			desc = r.group(2)
			if gpu % 2 == 0:
				self.cards.append(Card(card, []))
			self.cards[-1].gpus.append(GPU(gpuid, card, desc))
			print "gpu %d card %d desc %s" % (gpuid, card, desc)
			if gpu % 2 == 1:
				card = card + 1
			gpu = gpu + 1

	def watch(self):
		while True:
			for card in self.cards:
				if card is None:
					continue
				self.watch_card(card)
			time.sleep(settings['poll_time'])
			print "----------------------------------"

	def watch_card(self, card):
		fanspeed = card.fanspeed()
		fandelta = 0
		print "Card %d: fan speed %d%%" % (card.id, fanspeed)
		for gpu in card.gpus:
			temp = gpu.temp()
			print "Card %d GPU %d: Temperature %.2f °C" % (
				card.id, gpu.id, temp)
			if temp >= settings['temp_cooldown']:
				for pid in psutil.get_pid_list():
					try:
						p = psutil.Process(pid)
						if len(p.cmdline) > 2 and 'poclbm' in p.cmdline[1]:
							n = int(p.cmdline[2][2])
							if n == gpu.id:
								print "Suspending GPU %d for 10 seconds" % (n)
								os.kill(p.pid, signal.SIGSTOP)
								time.sleep(settings['cooldown_period'])
								os.kill(p.pid, signal.SIGCONT)
					except:
						pass
			if temp >= settings['temp_fan_100']:
				fandelta = 100 - fanspeed
				continue
			if temp > settings['temp_high']:
				if fanspeed <= 90:
					fandelta = +10
				else:
					fandelta = 100 - fanspeed
				continue
			elif temp < settings['temp_low']:
				if fanspeed <= 5:
					continue
				fandelta = -5
		if fandelta != 0 and fanspeed + fandelta >= 10:
			print "Card %d: Adjusting fan %d%% to %d%%" % (
				card.id, fandelta, fanspeed + fandelta)
			card.fanspeed(fanspeed + fandelta)

if __name__ == '__main__':
	if len(sys.argv) > 1:
		try:
			f = open(sys.argv[1])
			for line in f:
				m = re.search('^(\w+)\s*=\s*(\S.*)$', line)
				if m is None:
					continue
				settings[m.group(1)] = m.group(2)
			f.close()
		except (OSError, IOError), e:
			pass

	settings['poll_time'] = int(settings['poll_time'])
	settings['temp_low'] = float(settings['temp_low'])
	settings['temp_high'] = float(settings['temp_high'])
	print repr(settings)
	w = Watcher()
	w.watch()

#!/usr/bin/python
# -*- coding: utf-8 -*-

import sys
import time
import re
import os
import subprocess
import psutil
import signal
from string import split

settings = {
	'poll_time': 4,
	'fan_overTemp': 10,
	'temp_fan_100': 90.0,
	'temp_cooldown': 93.0,
	'cooldown_period': 10,
	'ati_card_exclude': '4250',
	'aticonfig_path': '/usr/bin/aticonfig'
}

class GPU():
	def __init__(self, id, cardid, description):
		self.id = id
		self.cardid = cardid
		self.description = description

	def temp(self):
		os.environ['DISPLAY'] = ":0"
		try:
			line = subprocess.check_output(
				[settings['aticonfig_path'],
				 "--adapter=" + str(self.id),
				 "--odgt"])
			m = re.search('Temperature - (\d+\.\d+)\s*C', line)
			if m is None:
				return None

			return float(m.group(1))
		except OSError:
			return None

class Card():
	def __init__(self, id, gpus):
		self.id = id
		self.x11_id = ':0.' + str(id)
		self.gpus = gpus

	def fanspeed(self, val=None):
		os.environ['DISPLAY'] = self.x11_id
		if val is not None:
			subprocess.call( [settings['aticonfig_path'],
				 "--pplib-cmd", "set fanspeed 0 " + str(val)])
		else:
			line = subprocess.check_output(
				[settings['aticonfig_path'],
				 "--pplib-cmd", "get fanspeed 0"])
			m = re.search('Fan Speed: (\d+)%', line)
			if m is None:
				return False
			return int(m.group(1))

class Watcher():
	def __init__(self):
		self.cards = []
		os.environ['DISPLAY'] = ":0"
		out = subprocess.check_output([settings['aticonfig_path'],
			"--list-adapters"])
		if settings['ati_card_exclude'] == '':
			card = 0
		else: card = 1
		gpu = 0
		lines = split(out, "\n")
		for line in lines:
			r = re.search('^[ \*]+(\d+)\. [\d:\.]+ (.+)$', line)
			if r is None:
				continue
			gpuid = int(r.group(1))
			desc = r.group(2)
			if settings['ati_card_exclude'] in desc:
				continue
			if gpu % 2 == 0:
				self.cards.append(Card(card, []))
			self.cards[-1].gpus.append(GPU(gpuid, card, desc))
			print "gpu %d card %d desc %s" % (gpuid, card, desc)
			if gpu % 2 == 1:
				card = card + 1
			gpu = gpu + 1

	def watch(self):
		while True:
			for card in self.cards:
				if card is None:
					continue
				self.watch_card(card)
			time.sleep(settings['poll_time'])
#			print "----------------------------------"

	def watch_card(self, card):
		fanspeed = card.fanspeed()
		fandelta = 0
		print "Card %d: fan speed %d%%" % (card.id, fanspeed)
		for gpu in card.gpus:
			temp = gpu.temp()
#			print "Card %d GPU %d: Temperature %.2f °C" % (
#				card.id, gpu.id, temp)
			if temp >= settings['temp_cooldown']:
				for pid in psutil.get_pid_list():
					try:
						p = psutil.Process(pid)
						if len(p.cmdline) > 2 and 'poclbm' in p.cmdline[1]:
							n = int(p.cmdline[2][2])
							if n == gpu.id:
								print "Suspending GPU %d for 10 seconds" % (n)
								os.kill(p.pid, signal.SIGSTOP)
								time.sleep(settings['cooldown_period'])
								os.kill(p.pid, signal.SIGCONT)
					except:
						pass
			if gpu.id % 2 != 0 :  # first gpu on card - usually the cooler one
				fanspeed2 = int(temp) + settings['fan_overTemp']
				continue #check the temp on the second gpu of this card
			else:  # second gpu on card
				fanspeed = int(temp) + settings['fan_overTemp']
			if fanspeed2 > fanspeed:   # use the highest speed
				fanspeed = fanspeed2
				if fanspeed > 100:
					fanspeed = 100
			card.fanspeed(fanspeed)

if __name__ == '__main__':
	if len(sys.argv) > 1:
		try:
			f = open(sys.argv[1])
			for line in f:
				m = re.search('^(\w+)\s*=\s*(\S.*)$', line)
				if m is None:
					continue
				settings[m.group(1)] = m.group(2)
			f.close()
		except (OSError, IOError), e:
			pass

	settings['poll_time'] = int(settings['poll_time'])
	print repr(settings)
	w = Watcher()
	w.watch()