go-nvidia

README

go-nvidia

Some wrappers of nvidia tookits for golang.

TODO

• cuda
• cublas
• tensorrt

Usage

# 1. if cuda is not installed in '/usr/local/cuda', please set CFLAGS and LDFLAGS like this:
export CGO_CFLAGS="-I/usr/local/cuda/include"
export CGO_LDFLAGS="-L/usr/local/cuda/lib64"
# 2.then you can run tests
go test github.com/CJH9004/nvidia/cuda
go test github.com/CJH9004/nvidia/cublas
go test github.com/CJH9004/nvidia/tensorrt