andr3/gotch
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

clean-up lib.go

Browse Source
This commit is contained in:
sugarme 2020-07-30 15:43:46 +10:00
parent 2fdd5a3284
commit 3f95b46f6c
2 changed files with 16 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
README.md
View File

@ -18,7 +18,17 @@
- **Libtorch** C++ v1.5.0 library of [Pytorch](https://pytorch.org/)
- Make sure that a libtorch version 1.5.0 (either CPU or CUDA support) is
installed in your system (default at "/opt/libtorch" in Linux/Mac OS).
installed in your system (at "/opt/libtorch" in Linux/Mac OS).
- Update `.bashrc` file to include `libtorch` in `LD_LIBRARY_PATH`
```bash
# .bashrc file
export LIBTORCH=/opt/libtorch
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/include:${LIBTORCH}:${LIBTORCH}/lib:${LIBTORCH}/include:${LIBTORCH}/include/torch/csrc/api/include
```
- Alternatively, just execute corresponding command to setup Libtorch C++

15
libtch/lib.go
View File

@ -1,15 +1,10 @@
package libtch
// #cgo CXXFLAGS: -std=c++17 -I${SRCDIR} -g -O3
// #cgo LDFLAGS: -lstdc++ -ltorch -lc10 -ltorch_cpu -L/opt/libtorch/lib
// #cgo LDFLAGS: -L/usr/local/cuda/lib64 -lcuda -lcudart -lcublas -lcudnn -lcaffe2_nvrtc -lnvrtc-builtins -lnvrtc -lnvToolsExt -lc10_cuda -ltorch_cuda
// #cgo CFLAGS: -I${SRCDIR} -O3 -Wall -Wno-unused-variable -Wno-deprecated-declarations -Wno-c++11-narrowing -g -Wno-sign-compare -Wno-unused-function
// #cgo CFLAGS: -I/usr/local/include -I/opt/libtorch/include -I/opt/libtorch/include/torch/csrc/api/include
// #cgo LDFLAGS: -lstdc++ -ltorch -lc10 -ltorch_cpu
// #cgo LDFLAGS: -L/opt/libtorch/lib -L/lib64
// #cgo CXXFLAGS: -isystem /opt/libtorch/lib
// #cgo CXXFLAGS: -isystem /opt/libtorch/include
// #cgo CXXFLAGS: -isystem /opt/libtorch/include/torch/csrc/api/include
// #cgo CXXFLAGS: -isystem /opt/libtorch/include/torch/csrc
// #cgo CFLAGS: -D_GLIBCXX_USE_CXX11_ABI=1
// #cgo linux,amd64,!nogpu CFLAGS: -I/usr/local/cuda/include
// #cgo linux,amd64,!nogpu LDFLAGS: -L/usr/local/cuda/lib64 -lcuda -lcudart -lcublas -lcudnn -lcaffe2_nvrtc -lnvrtc-builtins -lnvrtc -lnvToolsExt -L/opt/libtorch/lib -lc10_cuda -ltorch_cuda
// #cgo CFLAGS: -I/usr/local/cuda/include
// #cgo CXXFLAGS: -std=c++17 -I${SRCDIR} -g -O3
// #cgo CXXFLAGS: -I/opt/libtorch/lib -I/opt/libtorch/include -I/opt/libtorch/include/torch/csrc/api/include
import "C"