WIKIPEDIA
Definition: https://en.wikipedia.org/wiki/CUDA
- Papers used CUDA at conferences CVPR2022
-
.............
Content of this repository:
- CUDA Definition.
- CUDA with C++
- CUDA/C++ with Pytorch
Component | Description |
---|---|
torch | A Tensor library like NumPy, with strong GPU support |
torch.autograd | A tape-based automatic differentiation library that supports all differentiable Tensor operations in torch |
torch.jit | A compilation stack (TorchScript) to create serializable and optimizable models from PyTorch code |
torch.nn | A neural networks library deeply integrated with autograd designed for maximum flexibility |
torch.multiprocessing | Python multiprocessing, but with magical memory sharing of torch Tensors across processes. Useful for data loading and Hogwild training |
torch.utils | DataLoader and other utility functions for convenience |
This material is made for presentation purposes for the Group Report of NEFU ICEC