Run the following command to retrieve the source code into the directory $GOPATH/src/github.com/wangkuiyi/gotorch.

go get github.com/wangkuiyi/gotorch

Feel free to set the environment variable GOPATH to an arbitrary directory export GOPATH=some/directory.

Build the Cgo binding of libtorch.

$GOPATH/src/github.com/wangkuiyi/gotorch/cgotorch/build.sh

Run the Go examples and unit tests.

cd $GOPATH/src/github.com/wangkuiyi/gotorch
go test -v github.com/wangkuiyi/gotorch/...

The above go test command might fail and complain that it cannot find the .so or .dylib files. If so, please run the following command.

export CGOTORCH=$GOPATH/src/github.com/wangkuiyi/gotorch/cgotorch/libtorch/lib
export LD_LIBRARY_PATH=$CGOTORCH:$LD_LIBRARY_PATH