Ubuntu 22.04 x86_64 源码编译 pytorch-v2.0.1 笔记

**Ubuntu22.04 x86_64 源码编译 PyTorch-v2.0.1 笔记**

**前言**

PyTorch 是一个流行的开源机器学习库，提供了一个易用的接口来定义和执行计算图。源码编译是了解 PyTorch 内部工作原理的一种方式，也可以自定义和优化 PyTorch 的行为。

在本文中，我们将介绍如何在 Ubuntu22.04 x86_64 系统上源码编译 PyTorch-v2.0.1。

**环境准备**

* 操作系统：Ubuntu22.04 x86_64* Python 版本：Python3.9.x (建议使用最新版本)
* GCC 版本：GCC12.x (建议使用最新版本)

**安装依赖包**

首先，我们需要安装必要的依赖包：

bashsudo apt update && sudo apt install -y 
 build-essential 
 libncurses5-dev 
 libstdc++6 
 zlib1g-dev 
 libbz2-dev 
 libffi-dev 
 libssl-dev 
 libblas-dev 
 liblapack-dev 
 libatlas-base-dev 
 libopenblas-dev 
 libhdf5-serial-dev 
 libhdf5-dev 
 libprotobuf-dev 
 protobuf-compiler 
 libgoogle-glog-dev 
 libgflags-dev 
 libsnappy-dev 
 libthrift-dev 
 liblmdb-dev 
 libboost-all-dev 
 libcurl4-openssl-dev 
 libzmq3-dev

**下载源码**

接下来，我们需要下载 PyTorch 的源码：

bashgit clone  pytorchgit checkout v2.0.1

**配置和编译**

现在，我们可以开始配置和编译 PyTorch 了：

首先，创建一个 `build` 目录，并进入该目录：

bashmkdir build && cd build

然后，使用 CMake 来配置 PyTorch 的构建：

bashcmake ..

如果你想自定义构建选项，可以在此处添加 `-D`选项。例如，要禁用 CUDA 支持，可以使用以下命令：

bashcmake -DCUDA_ARCH=0 ..

最后，开始编译 PyTorch：

bashmake -j$(nproc)

**安装**

一旦编译完成，就可以安装 PyTorch 了：

bashsudo make install

**测试**

最后，我们需要测试一下 PyTorch 是否正确安装了：

import torchprint(torch.__version__)

如果输出为 `2.0.1`，则表示 PyTorch 已经正确安装。

**总结**

在本文中，我们介绍了如何在 Ubuntu22.04 x86_64 系统上源码编译 PyTorch-v2.0.1。我们一步步地完成了环境准备、依赖包安装、源码下载、配置和编译、安装以及测试等过程。如果你遇到任何问题，请尝试检查你的环境是否正确设置，或者参考 PyTorch 的官方文档。

**附注**

* 如果你想自定义构建选项，可以在 `CMake` 时使用 `-D`选项。
* 如果你想禁用 CUDA 支持，可以使用 `-DCUDA_ARCH=0`选项。
* 如果你想禁用 cuDNN 支持，可以使用 `-DCUDNN_LIBRARY=""`选项。

**参考**

* PyTorch 官方文档： CMake 官方文档：