I would recommend to update to the latest version and build PyTorch from source as described here 1.
This would not only include a lot of new features, but also bug fixes.
Let us know, if you get stuck or if this wouldn’t work for your setup.