1. 前置条件
本篇续接自 win11编译pytorch cuda128版本流程,阅读前请先参考上一篇配置环境。
访问https://kkgithub.com/pytorch/vision/archive/refs/tags/v0.21.0.tar.gz下载源码,下载后解压。
2.编译
打开Miniforge Prompt,依次执行如下代码:
conda activate torch_build_env
conda install libpng libjpeg-turbo -c pytorchcd c:\vision-0.21.0
##执行前先确认删除build目录
python setup.py bdist_wheel
编译完成后,wheel包在dist目录下,使用pip安装
pip install dist\torchvision-0.21.0-cp310-cp310-win_amd64.whl