为 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 安装和编译项目的好文档。有认识的大神吗？

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

Answer 1

我用下一个安装在Ubuntu16上，对你有帮助。

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

也可以参考官方文档使用PyTorch c++ for Linux systems and the GCPdocumentation。