为ubuntu安装pytorch c++ api CUDA11.4

Question

我正在尝试在 ubuntu 18.04 上使用 PyTorch c++ API。我已经安装了 CUDA 11.4 和 cuDNN 8.2.4.15。我正在编译的源代码可在此处获得。使用 nvcc 编译 CUDA 工作，并且 cuDNN 安装测试成功。但是我找不到在 Ubuntu 上使用 PyTorch c++ api 安装和编译项目的好文档。你认识神吗？

系统配置：
操作系统：ubuntu 18.04
GPU：1 x NVIDIA Tesla P4
机器类型：谷歌云控制台上的 n1-standard-2（2 个 vCPU，7.5 GB 内存）
CUDA：11.4
cuDNN：8.2.4.15

Answer 1

我使用了下一个安装在 Ubuntu 16 上，对你有帮助。

• PyTorch C++ API Ubuntu Installation Guide
• tutorial to compile and use pytorch on ubuntu 16.04

也可以参考官方文档使用PyTorch c++ for Linux系统和GCPdocumentation。