Skip to content

Commit

Permalink
update torch, torchvision, mmdet, mmcv in tests (#379)
Browse files Browse the repository at this point in the history
* update torch, torchvision, mmdet, mmcv in tests

* Update package_testing.yml

* update installation in readme
  • Loading branch information
fcakyon authored Feb 12, 2022
1 parent 84df247 commit fb0c9ba
Show file tree
Hide file tree
Showing 3 changed files with 11 additions and 11 deletions.
8 changes: 4 additions & 4 deletions .github/workflows/ci.yml
Original file line number Diff line number Diff line change
Expand Up @@ -75,12 +75,12 @@ jobs:
- name: Install PyTorch on MacOS
if: matrix.operating-system == 'macos-latest'
run: pip install torch==1.10.0 torchvision==0.11.1
run: pip install torch==1.10.2 torchvision==0.11.3

- name: Install MMDetection(2.20.0) with MMCV(1.4.2)
- name: Install MMDetection(2.20.0) with MMCV(1.4.4)
run: >
pip install mmcv-full==1.4.2 -f https://download.openmmlab.com/mmcv/dist/cpu/torch1.10.0/index.html
pip install mmdet==2.20.0
pip install mmcv-full==1.4.4 -f https://download.openmmlab.com/mmcv/dist/cpu/torch1.10.0/index.html
pip install mmdet==2.21.0
- name: Install YOLOv5(6.0.6) and Norfair(0.3.1)
run: >
Expand Down
8 changes: 4 additions & 4 deletions .github/workflows/package_testing.yml
Original file line number Diff line number Diff line change
Expand Up @@ -60,12 +60,12 @@ jobs:
- name: Install PyTorch on MacOS
if: matrix.operating-system == 'macos-latest'
run: pip install torch==1.10.0 torchvision==0.11.1
run: pip install torch==1.10.2 torchvision==0.11.3

- name: Install MMDetection(2.20.0) with MMCV(1.4.2)
- name: Install MMDetection(2.21.0) with MMCV(1.4.4)
run: >
pip install mmcv-full==1.4.2 -f https://download.openmmlab.com/mmcv/dist/cpu/torch1.10.0/index.html
pip install mmdet==2.20.0
pip install mmcv-full==1.4.4 -f https://download.openmmlab.com/mmcv/dist/cpu/torch1.10.0/index.html
pip install mmdet==2.21.0
- name: Install YOLOv5(6.0.6) and Norfair(0.3.1)
run: >
Expand Down
6 changes: 3 additions & 3 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -97,7 +97,7 @@ conda install -c conda-forge shapely
- Install your desired version of pytorch and torchvision:

```console
conda install pytorch=1.10.0 torchvision=0.11.1 cudatoolkit=11.3 -c pytorch
conda install pytorch=1.10.2 torchvision=0.11.3 cudatoolkit=11.3 -c pytorch
```

- Install your desired detection framework (yolov5):
Expand All @@ -109,11 +109,11 @@ pip install yolov5
- Install your desired detection framework (mmdet):

```console
pip install mmcv-full==1.4.2 -f https://download.openmmlab.com/mmcv/dist/cu113/torch1.10.0/index.html
pip install mmcv-full==1.4.4 -f https://download.openmmlab.com/mmcv/dist/cu113/torch1.10.0/index.html
```

```console
pip install mmdet==2.20.0
pip install mmdet==2.21.0
```

- Install your desired detection framework (detectron2):
Expand Down

0 comments on commit fb0c9ba

Please sign in to comment.