torch==2.1.0
Update docker files to use torch 2.1 Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>