Updated CHANGELOG, README and scripts for v0.3.0
This commit is contained in:
parent
ca8e51fad8
commit
d2ad0e7f5a
|
@ -41,3 +41,9 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
|
||||||
### Added
|
### Added
|
||||||
- Added drawing image label at `example/yolo` example
|
- Added drawing image label at `example/yolo` example
|
||||||
- Added some example images and README files for `example/yolo` and `example/neural-style-transfer`
|
- Added some example images and README files for `example/yolo` and `example/neural-style-transfer`
|
||||||
|
|
||||||
|
## [0.3.0]
|
||||||
|
|
||||||
|
### Changed
|
||||||
|
- Updated to Pytorch C++ APIs v1.7.0
|
||||||
|
- Switched back to `lib.AtoAddParametersOld` as the `ato_add_parameters` has not been implemented correctly. Using the updated API will cause optimizer stops working.
|
||||||
|
|
16
README.md
16
README.md
|
@ -5,32 +5,32 @@
|
||||||
|
|
||||||
- **GoTch** is a C++ Libtorch Go binding for developing and implementing deep learning projects in Go.
|
- **GoTch** is a C++ Libtorch Go binding for developing and implementing deep learning projects in Go.
|
||||||
- This package is to create a thin wrapper of Libtorch to make use of its tensor APIs and CUDA support while implementing as much idiomatic Go as possible.
|
- This package is to create a thin wrapper of Libtorch to make use of its tensor APIs and CUDA support while implementing as much idiomatic Go as possible.
|
||||||
- There are about **1129** auto-generated tensor APIs.
|
- There are about **1404** auto-generated tensor APIs.
|
||||||
|
|
||||||
## Dependencies
|
## Dependencies
|
||||||
|
|
||||||
- **Libtorch** C++ v1.5.0 library of [Pytorch](https://pytorch.org/)
|
- **Libtorch** C++ v1.7.0 library of [Pytorch](https://pytorch.org/)
|
||||||
|
|
||||||
|
|
||||||
## Installation
|
## Installation
|
||||||
|
|
||||||
- **CPU**
|
- **CPU**
|
||||||
|
|
||||||
Default values: `LIBTORCH_VER=1.5.1` and `GOTCH_VER=v0.1.7`
|
Default values: `LIBTORCH_VER=1.7.0` and `GOTCH_VER=v0.3.0`
|
||||||
|
|
||||||
```bash
|
```bash
|
||||||
go get -u github.com/sugarme/gotch@v0.1.7
|
go get -u github.com/sugarme/gotch@v0.3.0
|
||||||
bash ${GOPATH}/pkg/mod/github.com/sugarme/gotch@v0.1.7/setup-cpu.sh
|
bash ${GOPATH}/pkg/mod/github.com/sugarme/gotch@v0.3.0/setup-cpu.sh
|
||||||
|
|
||||||
```
|
```
|
||||||
|
|
||||||
- **GPU**
|
- **GPU**
|
||||||
|
|
||||||
Default values: `LIBTORCH_VER=1.5.1`, `CUDA_VER=10.1` and `GOTCH_VER=v0.1.7`
|
Default values: `LIBTORCH_VER=1.7.0`, `CUDA_VER=10.1` and `GOTCH_VER=v0.3.0`
|
||||||
|
|
||||||
```bash
|
```bash
|
||||||
go get -u github.com/sugarme/gotch@v0.1.7
|
go get -u github.com/sugarme/gotch@v0.3.0
|
||||||
bash ${GOPATH}/pkg/mod/github.com/sugarme/gotch@v0.1.7/setup-gpu.sh
|
bash ${GOPATH}/pkg/mod/github.com/sugarme/gotch@v0.3.0/setup-gpu.sh
|
||||||
|
|
||||||
```
|
```
|
||||||
|
|
||||||
|
|
|
@ -1,8 +1,8 @@
|
||||||
#!/bin/bash
|
#!/bin/bash
|
||||||
|
|
||||||
# Env
|
# Env
|
||||||
GOTCH_VERSION="${GOTCH_VER:-v0.2.0}"
|
GOTCH_VERSION="${GOTCH_VER:-v0.3.0}"
|
||||||
LIBTORCH_VERSION="${LIBTORCH_VER:-1.5.1}"
|
LIBTORCH_VERSION="${LIBTORCH_VER:-1.7.0}"
|
||||||
|
|
||||||
GOTCH="$GOPATH/pkg/mod/github.com/sugarme/gotch@$GOTCH_VERSION"
|
GOTCH="$GOPATH/pkg/mod/github.com/sugarme/gotch@$GOTCH_VERSION"
|
||||||
LIBTORCH="$GOPATH/pkg/mod/github.com/sugarme/gotch@$GOTCH_VERSION/libtch/libtorch"
|
LIBTORCH="$GOPATH/pkg/mod/github.com/sugarme/gotch@$GOTCH_VERSION/libtch/libtorch"
|
||||||
|
|
|
@ -1,7 +1,7 @@
|
||||||
#!/bin/bash
|
#!/bin/bash
|
||||||
|
|
||||||
GOTCH_VERSION="${GOTCH_VER:-v0.2.0}"
|
GOTCH_VERSION="${GOTCH_VER:-v0.3.0}"
|
||||||
LIBTORCH_VERSION="${LIBTORCH_VER:-1.5.1}"
|
LIBTORCH_VERSION="${LIBTORCH_VER:-1.7.0}"
|
||||||
CUDA_VERSION="${CUDA_VER:-10.1}"
|
CUDA_VERSION="${CUDA_VER:-10.1}"
|
||||||
CU_VERSION="${CUDA_VERSION//./}"
|
CU_VERSION="${CUDA_VERSION//./}"
|
||||||
|
|
||||||
|
|
12
setup.sh
12
setup.sh
|
@ -1,15 +1,15 @@
|
||||||
#!/bin/bash
|
#!/bin/bash
|
||||||
|
|
||||||
GOTCH_VERSION="${GOTCH_VER:-v0.2.1}"
|
export GOTCH_VERSION="${GOTCH_VER:-v0.3.0}"
|
||||||
LIBTORCH_VERSION="${LIBTORCH_VER:-1.7.0}"
|
export LIBTORCH_VERSION="${LIBTORCH_VER:-1.7.0}"
|
||||||
CUDA_VERSION="${CUDA_VER:-10.1}"
|
export CUDA_VERSION="${CUDA_VER:-10.1}"
|
||||||
CU_VERSION="${CUDA_VERSION//./}"
|
export CU_VERSION="${CUDA_VERSION//./}"
|
||||||
|
|
||||||
export GOTCH="$HOME/projects/sugarme/gotch"
|
export GOTCH="$HOME/projects/sugarme/gotch"
|
||||||
export LIBTORCH="$HOME/projects/sugarme/gotch/libtch/libtorch"
|
export LIBTORCH="$HOME/projects/sugarme/gotch/libtch/libtorch"
|
||||||
export LIBRARY_PATH="$LIBTORCH/lib"
|
export LIBRARY_PATH="$LIBTORCH/lib"
|
||||||
export CPATH="$CPATH:$LIBTORCH/lib:$LIBTORCH/include:$LIBTORCH/include/torch/csrc/api/include"
|
export CPATH="$LIBTORCH/lib:$LIBTORCH/include:$LIBTORCH/include/torch/csrc/api/include"
|
||||||
export LD_LIBRARY_PATH="$LD_LIBRARY_PATH:$LIBTORCH/lib:/usr/lib64-nvidia:/usr/local/cuda-${CUDA_VERSION}/lib64"
|
export LD_LIBRARY_PATH="$LIBTORCH/lib:/usr/lib64-nvidia:/usr/local/cuda-${CUDA_VERSION}/lib64"
|
||||||
|
|
||||||
sudo rm -rf $LIBTORCH
|
sudo rm -rf $LIBTORCH
|
||||||
sudo mkdir -p $LIBTORCH
|
sudo mkdir -p $LIBTORCH
|
||||||
|
|
Loading…
Reference in New Issue
Block a user