pytorch, torchscript, cuda extension
- Directory
/torchscript_tuto: torch model load to c++
/pytorch_source: analyzing PyTorch source
- environment
- Ubuntu 18.04.5 LTS
- x68_64
- GPU: TITAN X
- nvidia ver: 455.23.05
- cuda ver: 11.1
- pytorch ver: 1.9.0
- cudnn ver: 7