win11编译pytorchvision cuda128版本流程

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 pytorch


cd 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