I got this
>>> from torch.utils.data import Dataset, DataLoader, random_split
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
ImportError: cannot import name 'random_split'

error
It means my installation of torch exists some issue?
torchvision               0.2.1                    py35_0
python                    3.5.6                hc3d631a_0
cudatoolkit               9.0                  h13b8566_0
pytorch                   0.4.0