PyTorch源码编译(windows)

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

PyTorch源码编译(windows)_第1张图片

2.PyTorch用途与安装方法:

PyTorch源码编译(windows)_第2张图片

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

PyTorch源码编译(windows)_第3张图片

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

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

PyTorch源码编译(windows)_第4张图片
  1. 使用conda安装能用依赖:

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

依赖安装完成:

PyTorch源码编译(windows)_第5张图片

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

PyTorch源码编译(windows)_第6张图片

安装UV库

PyTorch源码编译(windows)_第7张图片

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

克隆完成:

编译安装:

PyTorch源码编译(windows)_第8张图片

编译中...

PyTorch源码编译(windows)_第9张图片

编译完成

PyTorch源码编译(windows)_第10张图片

现在可开始机器学习了

PyTorch源码编译(windows)_第11张图片

PyTorch源码编译(windows)_第12张图片

你可能感兴趣的:(Python,pytorch,深度学习,python)