因为工作原因,最近有接触到 PyTorch,今天整理一下在 Windows 平台上安装和配置 PyTorch 的过程。

如果网络环境好,尤其是可以流畅从国外网站下载安装包的话,可以按照 PyTorch 官方指导安装(pytorch.org/get-started)。

如果网络环境不好,可以像我下面的做法,手动下载各个安装包,然后本地安装。

从 PyTorch 官方的安装指导网页上可以看到,目前 PyTorch 的稳定版为 1.11.0 及对应的 CUDA 版本。

目前,CUDA 最新版本为 11.7。根据前面的版本选择,我找到了 CUDA 11.5.2 的下载页面。

因为网络环境不好,导致在线安装很大几率中断,所以下载安装包,本地安装可以很好的保证成功率。另外,手动下载安装包之后本地安装也给版本选择、搭配提供很大空间。