utils.py in references can't work with pytorch-cpu

[LXYTSOS]

These codes can work well when I use a GPU version pytorch, and today I train a model use a CPU version, I got AssertionError: Torch not compiled with CUDA enabled, and then I trace the code I found this code in utils.py:memory=torch.cuda.max_memory_allocated(), so I use torch.cuda.is_available() to fix it:

if torch.cuda.is_available():
    print(log_msg.format(
        i, len(iterable), eta=eta_string,
        meters=str(self),
        time=str(iter_time), data=str(data_time),
        memory=torch.cuda.max_memory_allocated() / MB))

[codecov-io]

Codecov Report

Merging #1023 into master will not change coverage.
The diff coverage is n/a.

@@           Coverage Diff           @@
##           master    #1023   +/-   ##
=======================================
  Coverage   63.29%   63.29%           
=======================================
  Files          65       65           
  Lines        5152     5152           
  Branches      772      772           
=======================================
  Hits         3261     3261           
  Misses       1667     1667           
  Partials      224      224

---

Continue to review full report at Codecov.

Legend - Click here to learn more
Δ = absolute <relative> (impact), ø = not affected, ? = missing data
Powered by Codecov. Last update 2694a5d...fca5139. Read the comment docs.

[soumith]

thanks!