0.2 branch does not contain all the necessary changes for cuda 9/volta. You are right, a lot of changes lately, some of them related to reinforcement learning. Talk to reinforcement people which commit is usable.
In particular, some StochasticFunctions were removed in this PR https://github.com/pytorch/pytorch/pull/3165