安装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 |
python |
1.10.2 |
0.11.3 |
>=3.6 |
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等。如下图所示