目录
安装pytorch不需要预装cuda,pytorch库的cuda版本本身就有显卡驱动。无论是win还是ubuntu都可使用下面的流程完成离线安装。这里说的离线安装是指先下载完整的pytorch的安装包,再安装pytorch。
一、下载和安装pytorch
根据python版本下载合适的cuda或cpu版本的whl文件。(下载cuda版本torch需要注意的是3080ti、3090等较新的的显卡仅支持高版本的cuda),一般下载最新的就行。
下载完成后直接pip install torch-xxx-cu-xxx.whl
,就可快速安装pyotrch。
pytorch、torchvision .whl各个版本文件下载地址
二、根据下载的pytorch版本,回到上面的whl文件下载地址去下载合适的torchvision版本。
下载完成后直接pip install torch-xxx-cu-xxx.whl
,就可快速安装torchvision。
以下是本人收集的部分torchvision版本与torch对应关系,如果本博客没有及时更新,请自行搜索“pytorch与torchvision版本对应关系”。如果搜索不到,请看步骤三。
torch与torchvision版本对应关系
torch 版本 | torchvision 版本 | 兼容的python版本范围 |
---|---|---|
2.2 | 0.17 | >=3.8, <=3.11 |
2.1 | 0.16 | >=3.8, <=3.11 |
2.0 | 0.15 | >=3.8, <=3.11 |
1.13 | 0.14 | >=3.7.2, <=3.10 |
1.12 | 0.13 | >=3.7, <=3.10 |
1.11 | 0.12 | >=3.7, <=3.10 |
1.10.2 | 0.11.3 | >=3.6 |
1.9 | 0.10 | >=3.6, <=3.9 |
1.8 | 0.9 | >=3.6, <=3.9 |
1.7 | 0.8 | >=3.6, <=3.9 |
1.6.0 | 0.7.0 | >=3.6 |
1.5.1 | 0.6.1 | >=3.5 |
1.5.0 | 0.6.0 | >=3.5 |
1.4.0 | 0.5.0 | ==2.7, >=3.5, <=3.8 |
1.3.1 | 0.4.2 | ==2.7, >=3.5, <=3.7 |
1.3.0 | 0.4.1 | ==2.7, >=3.5, <=3.7 |
1.2.0 | 0.4.0 | ==2.7, >=3.5, <=3.7 |
1.1.0 | 0.3.0 | ==2.7, >=3.5, <=3.7 |
<=1.0.1 | 0.2.2 | ==2.7, >=3.5, <=3.7 |
三、如果没有搜索到对应的torch版本也没关系。可以先安装torch后,再使用pip install torchvision
自动安装。
torchvision安装包较小很快就可以安装完,但使用这种方式可能会更新其他依赖包如numpy Pillow等。如下图所示