文章目录
- 1. 安装torch
- torch的下载链接
- 安装mmcv-full
- mmcv-full的下载链接
在安装mmcv-full中通常需要安装torch+mmcv-full。
1. 安装torch
在安装torch的时候,可以根据自身电脑是否有显卡,可以选择安装CPU版本还是GPU版本。mmcv-full也是同理。
安装torch和torchvision时,两个包都有版本对应关系。
torch | torchvision | python | cuda |
---|---|---|---|
2.2 | 0.17 | >=3.8, <=3.11 | |
2.1.0 | 0.16 | >=3.8, <=3.11 | 11.8, 12.1 |
2.0.0 | >0.14 | >=3.7, <=3.10 | 11.7, 11.8 |
1.13.0 | 0.14 | >=3.7, <=3.11 | 11.6, 11.7 |
1.12.0 | 0.13 | >=3.7, <=3.10 | 10.2, 11.3, 11.6 |
1.11.0 | 0.12.0 | >=3.7, <=3.10 | 10.2, 11.3 |
1.10.2 | 0.11.3 | >=3.6, <=3.9 | 10.2, 113 |
1.10.1 | 0.11.2 | >=3.6, <=3.9 | 10.2, 11.3 |
1.10.0 | 0.11.1 | >=3.6, <=3.9 | 20.2, 11.3 |
1.9.1 | 0.10.1 | >=3.6, <=3.9 | 10.2, 11.3 |
1.9.0 | 0.10.0 | >=3.6, <=3.9 | 10.2, 11.3 |
1.8.0 | 0.9.0 | >=3.6, <=3.9 | 10.2, 11.1 |
1.7.1 | 0.8.2 | >=3.6, <=3.9 | 9.2, 10.1, 10.2, 11.0 |
1.7.0 | 0.8.0 | >=3.6, <=3.8 | 9.2, 10.1, 10.2, 11.0 |
1.6.0 | 0.7.0 | >=3.6, <=3.8 | 9.2, 10.1, 10.2 |
1.5.1 | 0.6.1 | >=3.5, <=3.8 | 9.2, 10.1, 10.2 |
1.5.0 | 0.6.0 | >=3.5, <=3.8 | 9.2, 10.1, 10.2 |
1.4.0 | 0.5.0 | ==2.7, >=3.5, <=3.8 | 9.2, 10.0 |
1.3.1 | 0.4.2 | ==2.7, >=3.5, <=3.7 | 9.2, 10.0 |
1.3.0 | 0.4.1 | ==2.7, >=3.5, <=3.7 | 9.2, 10.0 |
1.2.0 | 0.4.0 | ==2.7, >=3.5, <=3.7 | 9.2, 10.0 |
1.1.0 | 0.3.0 | ==2.7, >=3.5, <=3.7 | 9.0, 10.0 |
<=1.0.1 | 0.2.2 | ==2.7, >=3.5, <=3.7 | 9.0, 10.0 |
torch的下载链接
不管是cpu的还是gpu的都在这里进行下载
在选择下载的时候,可根据自己的需要来选择
cpu
的包还是gpu
的包(这里选择的是cpu)- torch的
包的版本
是多少(这里选择的是1.4.0) - 是
python几
(这里选择的是python3.7) - 是
哪个系统
的包(这里是win系统) - 下载好torch也要下载
对应的
torchvision(根据表,torchvision要0.5.0的包)
安装mmcv-full
CUDA | torch 1.11 | torch 1.10 | torch 1.9 | torch 1.8 | torch 1.7 | torch 1.6 | torch 1.5 |
---|---|---|---|---|---|---|---|
11.5 | ✅ | ||||||
11.3 | ✅ | ✅ | |||||
11.1 | ✅ | ✅ | ✅ | ||||
11.0 | ✅ | ||||||
10.2 | ✅ | ✅ | ✅ | ✅ | ✅ | ✅ | ✅ |
10.1 | ✅ | ✅ | ✅ | ✅ | |||
9.2 | ✅ | ✅ | ✅ | ||||
cpu | ✅ | ✅ | ✅ | ✅ | ✅ | ✅ | ✅ |
以上表格表明存在该版本mmcv-full包
mmcv-full的下载链接
mmcv-full的下载链接:https://download.openmmlab.com/mmcv/dist/cu111/torch1.8.0/index.html
- 选择自己需要的 mmcv-full 版本、python 版本、操作系统 下的安装包
- 使用 pip install file-path 安装即可