GoTorch

Pytorch in Go, using LibTorch.

Build

Donwload LibTorch and extract it to the project directory.

LD_LIBRARY_PATH=libtorch/lib/ go run cmd/cuda_available.go

Name		Name	Last commit message	Last commit date
Latest commit History 11 Commits
cmd		cmd
.astylerc		.astylerc
.gitignore		.gitignore
CMakeLists.txt		CMakeLists.txt
LICENSE		LICENSE
README.md		README.md
cgo.go		cgo.go
cuda.cpp		cuda.cpp
cuda.go		cuda.go
cuda.h		cuda.h
device.cpp		device.cpp
device.go		device.go
device.h		device.h
go.mod		go.mod
gotorch.go		gotorch.go
nn.cpp		nn.cpp
nn.go		nn.go
nn.h		nn.h
tensor.cpp		tensor.cpp
tensor.go		tensor.go
tensor.h		tensor.h
tensor_options.h		tensor_options.h
version.go		version.go