PyTorch源码编译(windows)

1.打开pytorch源码仓库: ​​https://github.com/pytorch/pytorch#from-source​​

2.PyTorch用途与安装方法:

3.Python与编译器版本要求 (Python3.7或者更高,编译器要求支持C++17)

4.如果要支持CUDA编程,要安装NVIDIA CUDA 11或者更高版本, 安装NVIDIA cuDNN v7或者更高版本

CUDA不支持MacOS, 禁用CUDA ,设置 USE_CUDA=0

1. 使用conda安装能用依赖:

conda install astunparse numpy ninja pyyaml setuptools cmake cffi typing_extensions future six requests dataclasses

依赖安装完成:

根据要求安装下列包 (32位处理器才安装 : conda install mkl mkl-include ,64位处理器不用安装)

安装UV库

克隆PyTorch源码: git clone --recursive ​​https://github.com/pytorch/pytorch​​

克隆完成:

编译安装:

编译中...

编译完成

现在可开始机器学习了