Your PyTorch version is most likely pre 0.4.0.
Could you check it with:
import torch
print(torch.__version__)

I don’t know which version the fast.ai wrapper is using and I’m not sure, if the latest stable release is already implemented.
CC @jphoward