如何使用pip安装最新版TorchVision(0.14.0a0+b18fc23)

报错原因 #

直接执行pip3 install torchvision==0.14.0找不到包属于正常情况：0.14.0是未正式发布的开发迭代版本，不会上传到PyPI官方稳定源，pip默认检索稳定源时无法匹配到对应安装包。

正式发布的TorchVision 0.13.0版本未搭载面向动作识别任务的多尺度视觉Transformer（MViT）模型，该功能仅在0.14.0a0+b18fc23开发版本中上线，必须安装对应版本才可正常调用。

可用安装方式 #

方式一：通过PyTorch Nightly源安装（推荐，操作简便） #

PyTorch官方会将每日构建的开发版安装包托管在专属nightly源，安装时追加源地址即可拉取到开发版，对应命令如下：

• CUDA 11.7环境：

pip3 install --pre torch torchvision --extra-index-url https://download.pytorch.org/whl/nightly/cu117

• 纯CPU运行环境：

pip3 install --pre torch torchvision --extra-index-url https://download.pytorch.org/whl/nightly/cpu

如果使用其他CUDA版本，将命令末尾的cu117替换为对应版本号即可，比如CUDA11.6替换为cu116。

安装完成后执行以下命令校验版本：

python3 -c "import torchvision; print(torchvision.__version__)"

输出版本号为0.14.0a0开头即安装成功，可正常使用MViT相关接口。
注意：Nightly源为每日滚动更新版本，若需要精准锁定0.14.0a0+b18fc23这个对应commit的版本，使用下面的源码编译方案。

方式二：源码编译安装（精准匹配指定版本） #

操作步骤如下：

1. 拉取TorchVision代码仓库，切换到对应commit节点：

git clone https://github.com/pytorch/vision.git
cd vision
git checkout b18fc23

2. 安装编译所需依赖：

pip3 install numpy pillow requests

3. 执行编译安装：

python3 setup.py install

4. 同样执行上述版本校验命令，输出为0.14.0a0+b18fc23即安装完成。

内容的提问来源于stack exchange，提问作者Mike B