利用nvidia-dali进行训练加速时报错:
TypeError: current_stream() got an unexpected keyword argument ‘device’
解决办法:
一、升级pytorch版本
二、将安装路径中/python3.6/site-packages/nvidia/dali/plugin/pytorch.py文件中的
stream = torch.cuda.current_stream(device=pyt_tensors[category].device)
修改为:
stream = torch.cuda.current_stream()