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

updated installation script

Browse Source
This commit is contained in:
sugarme 2022-03-09 21:29:24 +11:00
parent e43f22df69
commit 44afdb50ec
2 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
setup-gotch.sh
View File

@ -2,6 +2,10 @@
GOTCH_VERSION="${GOTCH_VER:-v0.5.0}"
CUDA_VERSION="${CUDA_VER:-11.1}"
if [ -z $GOPATH ] then
$GOPATH="$HOME/go"
fi
GOTCH_PATH="$GOPATH/pkg/mod/github.com/sugarme/gotch@$GOTCH_VERSION"
# Install gotch

2
setup-libtorch.sh
View File

@ -3,7 +3,7 @@
LIBTORCH_VERSION="${LIBTORCH_VER:-1.10.0}"
CUDA_VERSION="${CUDA_VER:-11.1}"
if [[ -z "${CUDA_VERSION}"=="cpu" ]]; then
if [ "${CUDA_VERSION}"=="cpu" ]; then
CU_VERSION="cpu"
else
CU_VERSION="cu${CUDA_VERSION//./}"