I met the following error when running `radiology/train_base_nvtx.py`: 