pytorch cpp下载 - pytorch cpp源码下载

下载

适合所有人的 PyTorch 教程的 C++ 实现

入门

注意：交互式教程当前在LibTorch Nightly 版本上运行。
因此，有些教程在使用夜间版本时可能会中断。

conda create --name pytorch-cpp
conda activate pytorch-cpp
conda install xeus-cling notebook -c conda-forge

git clone https://github.com/prabhuomkar/pytorch-cpp.git
cd pytorch-cpp

cmake -B build # <options>

Windows 用户注意事项：
Libtorch 仅支持 64 位 Windows，需要指定 x64 生成器。对于 Visual Studio，可以通过将-A x64附加到上述命令来完成。

一些有用的选项：

选项	默认	描述
`-D CUDA_V=(11.8\|12.1\|none)`	`none`	下载 CUDA 版本的 LibTorch（ `none` = 下载 CPU 版本）。
`-D LIBTORCH_DOWNLOAD_BUILD_TYPE=(Release\|Debug)`	`Release`	确定要下载的 libtorch 构建类型版本（仅在Windows上相关）。
`-D DOWNLOAD_DATASETS=(OFF\|ON)`	`ON`	在构建过程中下载所需的数据集（仅当它们不存在于`pytorch-cpp/data`中时）。
`-D CREATE_SCRIPTMODULES=(OFF\|ON)`	`OFF`	在构建过程中为预先学习的模型/权重创建所有必需的脚本模块文件。需要安装 python3 以及 pytorch 和 torchvision。
`-D CMAKE_PREFIX_PATH=path/to/libtorch/share/cmake/Torch`	`<empty>`	跳过 LibTorch 的下载并使用您自己的本地版本（请参阅要求）。
`-D CMAKE_BUILD_TYPE=(Release\|Debug\|...)`	`<empty>`	确定单配置生成器的 CMake 构建类型（请参阅 CMake 文档）。

Linux 示例

cmake -B build 
-D CMAKE_BUILD_TYPE=Release 
-D CMAKE_PREFIX_PATH=/path/to/libtorch/share/cmake/Torch 
-D CREATE_SCRIPTMODULES=ON

示例窗口

cmake -B build 
-A x64 
-D CUDA_V=11.8

Windows (Visual Studio) 用户注意事项：
CMake 脚本下载 LibTorch 的Release版本，因此必须将--config Release附加到构建命令中。

数据集下载和脚本模块创建如何工作：

如果DOWNLOAD_DATASETS为ON ，您选择构建的教程所需的数据集将下载到pytorch-cpp/data （如果它们尚不存在）。
如果CREATE_SCRIPTMODULES为ON ，您选择构建的教程所需的预学习模型/权重的脚本模块文件将在相应教程源文件夹的model文件夹中创建（如果它们尚不存在）。

要构建所有教程，请使用

cmake --build build

您可以选择仅构建basics 、 intermediate 、 advanced或popular类别之一的教程。例如，如果您只对basics教程感兴趣：

cmake --build build --target basics
# In general: cmake --build build --target {category}

您还可以选择仅构建一个教程。例如仅构建语言模型教程：

cmake --build build --target language-model
# In general: cmake --build build --target {tutorial-name}

笔记：
目标参数是教程的文件夹名称，所有下划线都替换为连字符。

给 CMake 版本 >= 3.15 用户的提示：
您可以指定多个目标，并用空格分隔，例如：
cmake --build build --target language-model image-captioning

（重要！ ）首先切换到build/tutorials中的教程目录。例如，假设您位于pytorch-cpp目录中并想要更改为 pytorch 基础教程文件夹：
```
 cd build/tutorials/basics/pytorch_basics
# In general: cd build/tutorials/{basics|intermediate|advanced|popular/blitz}/{tutorial_name}
```
运行可执行文件。请注意，可执行文件的名称是教程的文件夹名称，所有下划线都替换为连字符（例如，对于教程文件夹： pytorch_basics -> 可执行文件名称： pytorch-basics （或 Windows 上的pytorch-basics.exe ））。例如，要运行 pytorch 基础教程：

Linux/苹果机
```
./pytorch-basics
# In general: ./{tutorial-name}
```
视窗
```
. pytorch-basics.exe
# In general: .{tutorial-name}.exe
```

在 Docker Hub 上查找最新和以前版本的映像。

您可以使用提供的Dockerfile和docker-compose.yml文件在 Docker 容器中构建并运行教程（在 CPU 上）：