【问题】在安装torchvision的时候，会自动安装torch？

1 背景

在使用如下命令安装torchvision的时候，发现之前已安装的torch被卸载了。

pip install torchvision -i https://pypi.tuna.tsinghua.edu.cn/simple

2 分析

在安装指定版本的torchvision后，pip总是会自动下载其他版本的torch。

3 解决方法

使用如的下命令进行安装就不会自动下载其他版本的torch。

pip install --no-deps torchvision==2.1.0

4 其他❤️❤️❤️

值得注意的是，torch和torchvision的版本需要对应。

版本对应说明

安装完成后，测试命令如下：

python3 -c "import torch" 

python3 -c "import torchvision"