Pytorch version check. This should be suitable for many users.