win11编译pytorch cuda128版本流程

Geforce 50xx系显卡最低支持cuda128，torch cu128 release版本目前还没有释放，所以自己基于2.6.0源码自己编译wheel包。

1. 前置条件

1. 使用visual studio installer 安装visual studio 2022，工作负荷选择【使用c++的桌面开发】,安装完成后将“VC\Tools\MSVC\<版本号>\bin\Hostx64\x64”对应的路径加入环境变量；

2. 访问https://github.com/pytorch/pytorch/releases/download/v2.6.0/pytorch-v2.6.0.tar.gz下载源码，可以替换成kkgithub镜像访问；

3. 访问https://github.com/conda-forge/miniforge/releases/download/24.11.3-0/Miniforge3-Windows-x86_64.exe安装miniforge；

2. 编译

打开 Miniforge Prompt，依次执行如下代码：

##如果访问报错或速度缓慢，搜索并替换conda、pip清华源
conda create torch_build_env python=3.10 libuv
conda activate torch_build_env
conda install -y astunparse numpy ninja pyyaml setuptools cmake cffi typing_extensions future six requests dataclasses

set MAX_JOBS=24
set USE_DISTRIBUTED=OFF
set USE_CUDA=ON
set PYTORCH_BUILD_VERSION=2.6.0a0+cu128-cp31016-win_x64
set PYTORCH_BUILD_NUMBER=0 


##执行前先确保删除build目录
python setup.py bdist_wheel

编译完成后，wheel包在dist目录下，使用pip安装

pip install dist\torch-2.6.0a0+cu128.cp31016.win.x64-cp310-cp310-win_amd64.whl