Pytorch-Extension-Cuda

A simple C++ / CUDA extension demo in pytorch which support multiplied GPUs
一个支持多GPU的pytorch C++ / CUDA 扩展示例

This repo was edited on vscode and you can change the paths in .vscode/c_cpp_properties.json to config intellisense
你可以通过改变 .vscode/c_cpp_properties.json里的路径配置使得代码提示功能正常运行

Installation notes / 安装

pytorch >= 2.0.0

git clone https://github.com/debugdoctor/pytorch-extension-cuda.git

cd pytorch-extension-cuda

pip install .

All setup configurations are in setup.py / 安装的配置可在setup.py中修改