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

update CHANGELOG.md and setup scripts

Browse Source
This commit is contained in:
sugarme 2021-07-23 11:01:39 +10:00
parent 842dd45f52
commit 6734fdb6f9
4 changed files with 10 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
CHANGELOG.md
View File

@ -11,6 +11,9 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
## [Nofix]
- ctype `long` caused compiling error in MacOS as noted on [#44]. Not working on linux box.
## [0.4.0]
- **Update libtorch to 1.9**. Generated **1716 APIs**. There are APIs naming changes ie. `Name1` change to `NameDim` or `NameTensor`.
## [0.3.14]
- Fixed temporary fix huge number of learning group returned from C at `libtch/tensor.go AtoGetLearningRates`
- Fixed incorrect `nn.AdamWConfig` and some documentation.

10
README.md
View File

@ -18,12 +18,12 @@ Gotch is in active development mode and may have API breaking changes. Feel free
## Dependencies
- **Libtorch** C++ v1.7.0 library of [Pytorch](https://pytorch.org/)
- **Libtorch** C++ v1.9.0 library of [Pytorch](https://pytorch.org/)
## Installation
- Default CUDA version is `10.1` if CUDA is available otherwise using CPU version.
- Default Pytorch C++ API version is `1.7.0`
- Default Pytorch C++ API version is `1.9.0`
**NOTE**: `libtorch` will be installed at **`/usr/local/lib`**
@ -51,7 +51,7 @@ Gotch is in active development mode and may have API breaking changes. Feel free
```bash
wget https://raw.githubusercontent.com/sugarme/gotch/master/setup-gotch.sh
chmod +x setup-gotch.sh
export CUDA_VER=cpu && export GOTCH_VER=v0.3.14 && bash setup-gotch.sh
export CUDA_VER=cpu && export GOTCH_VER=v0.4.0 && bash setup-gotch.sh
```
### GPU
@ -89,9 +89,9 @@ Gotch is in active development mode and may have API breaking changes. Feel free
wget https://raw.githubusercontent.com/sugarme/gotch/master/setup-gotch.sh
chmod +x setup-gotch.sh
# CUDA 10.1
export CUDA_VER=10.1 && export GOTCH_VER=v0.3.14 && bash setup-gotch.sh
export CUDA_VER=10.1 && export GOTCH_VER=v0.4.0 && bash setup-gotch.sh
# CUDA 11.0
export CUDA_VER=11.0 && export GOTCH_VER=v0.3.14 && bash setup-gotch.sh
export CUDA_VER=11.0 && export GOTCH_VER=v0.4.0 && bash setup-gotch.sh
```
## Examples

2
setup-gotch.sh
View File

@ -1,6 +1,6 @@
#!/bin/bash
GOTCH_VERSION="${GOTCH_VER:-v0.3.14}"
GOTCH_VERSION="${GOTCH_VER:-v0.4.0}"
CUDA_VERSION="${CUDA_VER:-10.1}"
GOTCH_PATH="$GOPATH/pkg/mod/github.com/sugarme/gotch@$GOTCH_VERSION"

2
setup-libtorch.sh
View File

@ -1,6 +1,6 @@
#!/bin/bash
LIBTORCH_VERSION="${LIBTORCH_VER:-1.7.0}"
LIBTORCH_VERSION="${LIBTORCH_VER:-1.9.0}"
CUDA_VERSION="${CUDA_VER:-10.1}"
if [[ -z "${CUDA_VERSION}"=="cpu" ]]; then