[DOCKER] Pin torchvision==0.4.1 (#4140)
authorMarcus Shawcroft <marcus.shawcroft@arm.com>
Thu, 17 Oct 2019 17:04:46 +0000 (18:04 +0100)
committerTianqi Chen <tqchen@users.noreply.github.com>
Thu, 17 Oct 2019 17:04:46 +0000 (10:04 -0700)
commita8a983176dfe41506458b628e7666dd0a6347807
tree4b7304c7b092c2f146d454fe97cd2cb4269f087d
parent972f019c73cb0118e6d5b9ecf1f8209e5660e20d
[DOCKER] Pin torchvision==0.4.1 (#4140)

The existing sequence of pip install commands fetches and installs
torch==1.0.1.post2 then fetches an unpinned version of torchvision,
recent torchvision packages hardwire the specific torch version they
depend on, the overall effect is that we install a pinned torch
version then replace it with whatever version the torchvision package
depends on.

The most recent torchvision==0.4.1 package results in some test case
failures.

This patch pins torchvision back to 0.4.0, the most recent version
that the test suite worked.  Removing the explicit torch install
because it is implied and pinned as dependency of torchvision.

Change-Id: Ib30bf6aed79ff130ea15ef5134fefb0508790574
docker/install/ubuntu_install_onnx.sh