如何下载安装包进行 PyTorch 和 torchvision 的离线安装

如何下载安装包进行 PyTorch 和 torchvision 的离线安装

在没有互联网连接的环境下安装 PyTorch 和 torchvision 可能看起来有点困难，但通过事先下载所需的安装包，我们可以轻松地在任何系统上进行离线安装。本文将详细介绍如何下载并离线安装 PyTorch 和 torchvision。

1. 安装 PyTorch

要离线安装 PyTorch，首先需要从官方网站下载对应的 Wheel 文件（.whl）。以下是详细的步骤：

下载安装包：

1. 访问 PyTorch 的 Wheel 文件下载页面：PyTorch Wheel Downloads
2. 根据您的系统配置（如 Python 版本、CUDA 版本）选择合适的安装包版本。如果您不确定哪个版本适合您的系统，可以参考附录中的版本对应表。
3. 下载对应的 .whl 文件到您的本地机器或移动存储设备上。

安装步骤：

1. 确保您的系统已安装 Python 和 pip。如果未安装，需要先进行安装。
2. 使用命令行工具（如终端或命令提示符），导航到包含下载的 .whl 文件的目录。
3. 执行以下命令来安装 PyTorch：

   pip install torch-版本号.whl
   

   请将 torch-版本号.whl 替换为您下载的文件名。

2. 安装 torchvision

与 PyTorch 类似，torchvision 也可以通过预下载的 Wheel 文件进行离线安装。

下载安装包：

1. 访问 torchvision 的 Wheel 文件下载页面：torchvision Wheel Downloads
2. 选择与您的 PyTorch 版本和系统配置相匹配的 torchvision 版本。
3. 下载 .whl 文件到您的本地机器或移动存储设备上。

安装步骤：

1. 确保已正确安装 PyTorch，因为 torchvision 依赖于 PyTorch。

2. 在包含下载的 .whl 文件的目录中打开命令行工具。

3. 执行以下命令来安装 torchvision：

   pip install torchvision-版本号.whl
   

   替换 torchvision-版本号.whl 为您的文件名。

4. 安装完成后，你可以通过以下命令验证安装是否成功：

   python -c "import torch; print(torch.__version__)"
   

这个命令会输出安装的 PyTorch 版本，以确保安装正确。

如果遇到依赖问题，可能需要提前安装一些额外的包，可以根据提示执行相关的 pip install 命令来解决。

附录：torch 和 torchvision 版本对应表

确保您下载的 PyTorch 和 torchvision 版本相互兼容是非常重要的。以下是一些常见的版本对应关系，以帮助您选择合适的文件：

PyTorch 版本	torchvision 版本	Python 版本	支持的 CUDA 版本
1.8.1	0.9.1	3.6, 3.7, 3.8	10.2, 11.1
1.8.0	0.9.0	3.6, 3.7, 3.8	10.2, 11.1
1.7.1	0.8.2	3.6, 3.7, 3.8	9.2, 10.1, 10.2
1.7.0	0.8.1	3.6, 3.7, 3.8	9.2, 10.1, 10.2
1.6.0	0.7.0	3.6, 3.7, 3.8	9.2, 10.1, 10.2
1.5.1	0.6.1	3.5, 3.6, 3.7, 3.8	9.2, 10.1
1.5.0	0.6.0	3.5, 3.6, 3.7, 3.8	9.2, 10.1
1.4.0	0.5.0	3.5, 3.6, 3.7, 3.8	9.2, 10.1
…	…	…	…

通过以上步骤，即使在没有网络连接的情况下，您也可以成功安装 PyTorch 和 torchvision。这为在各种环境中部署机器学习模型提供了极大的灵活性和便利。