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

updated travis config

Browse Source
This commit is contained in:
sugarme 2020-10-12 20:43:53 +11:00
parent b279e18c5d
commit a0c93f4b8d
2 changed files with 36 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

40
.travis.yml
View File

@ -14,16 +14,36 @@ dist: bionic
before_install: before_install:
- sudo apt-get install clang-tools-9 - sudo apt-get install clang-tools-9
- wget -O /tmp/libtorch-cxx11-abi-shared-with-deps-1.5.1+cpu.zip https://download.pytorch.org/libtorch/cpu/libtorch-cxx11-abi-shared-with-deps-1.5.1%2Bcpu.zip
- unzip /tmp/libtorch-cxx11-abi-shared-with-deps-1.5.1+cpu.zip -d /opt - GOTCH_VERSION="${GOTCH_VER:-v0.1.7}"
- export LIBTORCH=/opt/libtorch - LIBTORCH_VERSION="${LIBTORCH_VER:-1.5.1}"
- export LD_LIBRARY_PATH=${LIBTORCH}/lib:$LD_LIBRARY_PATH - GOTCH="$GOPATH/pkg/mod/github.com/sugarme/gotch@$GOTCH_VERSION"
- printenv - LIBTORCH="$GOPATH/pkg/mod/github.com/sugarme/gotch@$GOTCH_VERSION/libtch/libtorch"
- ls - LIBRARY_PATH="$LIBTORCH/lib"
- rm libtch/dummy_cuda_dependency.cpp - CPATH="$LIBTORCH/lib:$LIBTORCH/include:$LIBTORCH/include/torch/csrc:$LIBTORCH/include/torch/csrc/api/include"
- mv libtch/fake_cuda_dependency.cpp.cpu libtch/fake_cuda_dependency.cpp - LD_LIBRARY_PATH="$LIBTORCH/lib"
- rm libtch/lib.go
- mv libtch/lib.go.cpu libtch/lib.go # Precompiled libtorch
- sudo rm -rf $LIBTORCH
- sudo mkdir -p $LIBTORCH
- wget -O /tmp/libtorch-cxx11-abi-shared-with-deps-${LIBTORCH_VERSION}+cpu.zip https://download.pytorch.org/libtorch/cpu/libtorch-cxx11-abi-shared-with-deps-${LIBTORCH_VERSION}%2Bcpu.zip
- sudo unzip /tmp/libtorch-cxx11-abi-shared-with-deps-${LIBTORCH_VERSION}+cpu.zip -d $GOTCH/libtch
# Update .bashrc
- FILE="$HOME/.bashrc"
- LN_LIBTORCH="export LIBTORCH=$GOPATH/pkg/mod/github.com/sugarme/gotch@$GOTCH_VERSION/libtch/libtorch"
- LN_LIBRARY_PATH="export LIBRARY_PATH=$LIBTORCH/lib"
- LN_CPATH="export CPATH=$LIBTORCH/lib:$LIBTORCH/include:$LIBTORCH/include/torch/csrc:$LIBTORCH/include/torch/csrc/api/include"
- LN_LD_LIBRARY_PATH="export LD_LIBRARY_PATH=$LIBTORCH/lib:$LD_LIBRARY_PATH"
- sudo grep -xqF -- "$LN_LIBTORCH" "$FILE" || sudo echo "$LN_LIBTORCH" >> "$FILE"
- sudo grep -xqF -- "$LN_LIBRARY_PATH" "$FILE" || sudo echo "$LN_LIBRARY_PATH" >> "$FILE"
- sudo grep -xqF -- "$LN_CPATH" "$FILE" || sudo echo "$LN_CPATH" >> "$FILE"
- sudo grep -xqF -- "$LN_LD_LIBRARY_PATH" "$FILE" || sudo echo "$LN_LD_LIBRARY_PATH" >> "$FILE"
- sudo rm $GOTCH/libtch/lib.go
- sudo cp $GOTCH/libtch/lib.go.cpu $GOTCH/libtch/lib.go
- sudo mv $GOTCH/libtch/dummy_cuda_dependency.cpp $GOTCH/libtch/dummy_cuda_dependency.cpp.gpu
- sudo mv $GOTCH/libtch/fake_cuda_dependency.cpp.cpu $GOTCH/libtch/fake_cuda_dependency.cpp
- source "$FILE"
script: script:
- go test -v github.com/sugarme/gotch/tensor - go test -v github.com/sugarme/gotch/tensor
- go test -v github.com/sugarme/gotch/nn - go test -v github.com/sugarme/gotch/nn

12
README.md
View File

@ -15,21 +15,21 @@
- **CPU** - **CPU**
Default values: `LIBTORCH_VER=1.5.1` and `GOTCH_VER=v0.1.5` Default values: `LIBTORCH_VER=1.5.1` and `GOTCH_VER=v0.1.7`
```bash ```bash
go get -u github.com/sugarme/gotch@v0.1.5 go get -u github.com/sugarme/gotch@v0.1.7
bash ${GOPATH}/pkg/mod/github.com/sugarme/gotch@v0.1.5/setup-cpu.sh bash ${GOPATH}/pkg/mod/github.com/sugarme/gotch@v0.1.7/setup-cpu.sh
``` ```
- **GPU** - **GPU**
Default values: `LIBTORCH_VER=1.5.1`, `CUDA_VER=10.1` and `GOTCH_VER=v0.1.5` Default values: `LIBTORCH_VER=1.5.1`, `CUDA_VER=10.1` and `GOTCH_VER=v0.1.7`
```bash ```bash
go get -u github.com/sugarme/gotch@v0.1.5 go get -u github.com/sugarme/gotch@v0.1.7
bash ${GOPATH}/pkg/mod/github.com/sugarme/gotch@v0.1.5/setup-gpu.sh bash ${GOPATH}/pkg/mod/github.com/sugarme/gotch@v0.1.7/setup-gpu.sh
``` ```