add nvidia extension for pyrit

Extension for pyrit to use nvidia gpu