通过 CLion 在 WSL2 的 CUDA 环境下使用 libtorch
# 说明
本文记录使用 CLion 的 IDE 环境,使用 WSL2 环境下的 CUDA 来调用 PyTorch 的 libtorch 库内容。
libtorch 是 PyTorch 框架提供的 C++ 接口,因为最近的课题对性能有比较高的要求,并且有自定义算子的需求,所以需要用到相关的代码。
WSL2 下的 CUDA 环境配置可以参考这篇文章。
# 配置
# 下载 libtorch
从 PyTorch 的官方网站下载自己所需版本的 libtorch 。注意和自己的 CUDA 版本也要对应。
# CLion 配置
打开 CLion 新建一个 CUDA executable 工程。然后修改...
more...