Sets up default torch environment