【问题】在安装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"