Skip to content

Commit b7431f4

Browse files
mshawcroftwweic
authored andcommitted
[DOCKER] Pin torchvision==0.4.1 (apache#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
1 parent 2f9be03 commit b7431f4

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

docker/install/ubuntu_install_onnx.sh

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -27,5 +27,4 @@ pip3 install onnx==1.5.0
2727
# not expose that in the wheel!!!
2828
pip3 install future
2929

30-
pip3 install https://download.pytorch.org/whl/cu80/torch-1.0.1.post2-cp36-cp36m-linux_x86_64.whl
31-
pip3 install torchvision
30+
pip3 install torch==1.2.0 torchvision==0.4.0

0 commit comments

Comments
 (0)