TypeError: _cuda() got an unexpected keyword argument ‘non_blocking’

Part 1 (2018) Beginner (2018)
2

replace non_blocking with async, non_blocking for PyTorch since version 0.4.0. CUDA syntax error