https://github.com/vfdev-5/pth-dispatcher-playground
PyTorch dispatcher playground
https://github.com/vfdev-5/pth-dispatcher-playground
Last synced: 4 months ago
JSON representation
PyTorch dispatcher playground
- Host: GitHub
- URL: https://github.com/vfdev-5/pth-dispatcher-playground
- Owner: vfdev-5
- Created: 2020-10-13T12:07:01.000Z (over 5 years ago)
- Default Branch: master
- Last Pushed: 2020-10-13T14:01:03.000Z (over 5 years ago)
- Last Synced: 2025-10-09T19:34:29.936Z (8 months ago)
- Language: CMake
- Size: 5.86 KB
- Stars: 1
- Watchers: 1
- Forks: 0
- Open Issues: 0
-
Metadata Files:
- Readme: README.md
Awesome Lists containing this project
README
# PyTorch Dispatcher playground
- `libtv` mimics version of torchvision's C++ frontend
- `tvapp`, application using `libtv`
## Build in docker
- Run container:
```
docker run -it --network=host -v $PWD:/cpp -w /cpp --shm-size 16G --security-opt seccomp:unconfined pytorch/pytorch:1.6.0-cuda10.1-cudnn7-devel /bin/bash
```
- Install CMake
```
sh install_cmake.sh
```
- Build `libtv`
```
export TORCH_PATH=$(dirname $(python -c "import torch; print(torch.__file__)"))
mkdir -p /cpp/libtv/bld && cd /cpp/libtv/bld
cmake .. -DTorch_DIR=$TORCH_PATH/share/cmake/Torch -DWITH_CUDA=on
make install
```
- Build `tvapp`
```
export TORCH_PATH=$(dirname $(python -c "import torch; print(torch.__file__)"))
mkdir -p /cpp/tvapp/bld && cd /cpp/tvapp/bld
cmake .. -DTorch_DIR=$TORCH_PATH/share/cmake/Torch -DWITH_CUDA=on
make
```