Would be great if you set the versions of each package on the requirements.txt file. Example: ``` torch==1.2.0 torchvision==0.4.0 ``` As it is it will try to get the last version of the packages, but these may not work properly in the long term.