@jyotish Downgrading to 1.3.x works locally, thanks.
How do you make sure it installs 1.3.x when submitting? I added torch==1.3.1 to requirements.txt (since RUN pip install -r requirements.txt --no-cache-dir is called in Dockerfile) but print(torch.__version__) in impala agent still gave me 1.5.x.