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

updated scripts

Browse Source
This commit is contained in:
sugarme 2020-10-29 23:46:40 +11:00
parent ce0eef2076
commit 59ea5f0e1b
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
setup-cpu.sh
View File

@ -1,7 +1,7 @@
#!/bin/bash
# Env
GOTCH_VERSION="${GOTCH_VER:-v0.1.9}"
GOTCH_VERSION="${GOTCH_VER:-v0.1.10}"
LIBTORCH_VERSION="${LIBTORCH_VER:-1.5.1}"
GOTCH="$GOPATH/pkg/mod/github.com/sugarme/gotch@$GOTCH_VERSION"

2
setup-gpu.sh
View File

@ -1,6 +1,6 @@
#!/bin/bash
GOTCH_VERSION="${GOTCH_VER:-v0.1.9}"
GOTCH_VERSION="${GOTCH_VER:-v0.1.10}"
LIBTORCH_VERSION="${LIBTORCH_VER:-1.5.1}"
CUDA_VERSION="${CUDA_VER:-10.1}"
CU_VERSION="${CUDA_VERSION//./}"