Pytorch CUDA CPP简易教程，在Windows上操作，Linux上同理_clion 安装pytorch

---

前言

这学期确定了研究方法，具体为三维重建相关，转而研究三维重建相关的知识。最近3D Gaussian Splatting方法效果十分的好，并且开源了源代码，因此十分值得对其源码进行研究，源码中对于可微光栅化的实现是基于CUDA实现的，因此想要后续对这块内容进行改进，则必须了解CUDA程序是如何编写的，作为一名DL农工，学习CUDA编程也有助于自己对于整个DL流程的理解，也能够进一步拓展自己的编程技术，因此无论出于什么目的，作为DL农工，我认为有必要掌握这项技术。

本教程仅作为Pytorch CUDA CPP开发的简易教程，我也是初学者。同时本教程的目的也是为了方便不熟悉CUDA项目结构的人快速了解一份新的程序的文件结构是怎样的。

一、使用的工具

Pytorch, libtorch, CUDA, C++, CLion, Pycharm

二、学习资源分享

如果想要详细了解整个CUDA程序的工作原理，可以观看这个详细教程：CUDA编程基础入门系列

如果想功利点的学习如何使用Pytorch调用CUDA程序，可以观看这个教程：Pytorch+cpp/cuda extension 教学

本简易教程也是基于第二个视频教程写的，这个教程讲了一种通用的Pytorch、CUDA程序编写范式，使用pytorch+cpp+cuda开发的几乎都遵循这套代码编写流程，因此学会怎么写，基本上就能看明白其他人的CUDA程序文件是如何组织的，尽管你还是有可能看不懂CUDA代码，但是你学完后一定能看明白CUDA程序是如何组织的。

自定义 C++ 和 CUDA 扩展
PYTORCH C++ API

三、libtorch环境配置

注意：我所有的操作都是在Windows上进行的，Linux上的操作应该也大差不差
libtorch环境配置的教程有很多，你可以参考其他人的博客进行配置，这里我仅贴我使用CLion配置libtorch的流程

1.配置CUDA、nvcc、cudnn

该项不进行说明，有许多攻略可供参考，这里贴一篇博客 Windows安装CUDA及cuDNN【保姆级教程】
注意安装的CUDA版本不要太新了，且要与libtorch提供的CUDA版本保持一致，目前官网提供的libtorch CUDA版本为11.8，因此你安装的CUDA最好也是11.8，否则会有许多奇怪的错误
你都来看CUDA编程了，不至于DL环境还不会配置吧