Is much of code to use has changed from 0.4 to 1?

Hi, I wonder if I am getting the following error because the code has changed from 0.4 to 1 or maybe because server error (due to unresolved packages)? I have no problem running it on my machine but after loading pytorch on server I get this error (which has PyTorch 1).

[jalal@scc-c12 src]$ module load torchvision
ModuleCmd_Load.c(208):ERROR:105: Unable to locate a modulefile for 'torchvision'
NameError                                 Traceback (most recent call last)
<ipython-input-3-0b0ce4a30f28> in <module>()
     29 # should I use   transforms.RandomRotation(10) instead?
     30 data_transforms = {
---> 31     'train': transforms.Compose([
     32         transforms.RandomResizedCrop(224),
     33         transforms.RandomHorizontalFlip(),

NameError: name 'transforms' is not defined

This is the version I have on my own machine:

torch.__version__

'0.4.1.post2'

Can you import the transforms as follows:

from torchvision import transforms

Then there should not be any problem.