From 182ba2369552332c7ed1b2e7d3966b76fd8de151 Mon Sep 17 00:00:00 2001 From: Rosie Wood Date: Wed, 23 Oct 2024 08:32:50 +0100 Subject: [PATCH] check torch installation --- .github/workflows/mr_ci_text_spotting.yml | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/.github/workflows/mr_ci_text_spotting.yml b/.github/workflows/mr_ci_text_spotting.yml index dbecbd56..52a96ff3 100644 --- a/.github/workflows/mr_ci_text_spotting.yml +++ b/.github/workflows/mr_ci_text_spotting.yml @@ -50,7 +50,7 @@ jobs: - name: Build and install run: | - CC=clang CXX=clang++ python -m pip install 'git+https://github.com/facebookresearch/detectron2.git' + python -m pip install 'git+https://github.com/facebookresearch/detectron2.git' python -m detectron2.utils.collect_env python -m pip install ".[dev]" @@ -107,6 +107,7 @@ jobs: python -m pip install wheel ninja opencv-python-headless onnx pytest-xdist python -m pip install numpy==1.26.4 python -m pip install torch==2.2.2 torchvision==0.17.2 -f https://download.pytorch.org/whl/torch_stable.html + python -c "import torch; print(torch.__version__)" # install from github to get latest; install iopath first since fvcore depends on it python -m pip install -U 'git+https://github.com/facebookresearch/iopath' python -m pip install -U 'git+https://github.com/facebookresearch/fvcore'