Script tweak to restrict cuda versio