updated installation script
This commit is contained in:
parent
e43f22df69
commit
44afdb50ec
|
@ -2,6 +2,10 @@
|
||||||
|
|
||||||
GOTCH_VERSION="${GOTCH_VER:-v0.5.0}"
|
GOTCH_VERSION="${GOTCH_VER:-v0.5.0}"
|
||||||
CUDA_VERSION="${CUDA_VER:-11.1}"
|
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"
|
GOTCH_PATH="$GOPATH/pkg/mod/github.com/sugarme/gotch@$GOTCH_VERSION"
|
||||||
|
|
||||||
# Install gotch
|
# Install gotch
|
||||||
|
|
|
@ -3,7 +3,7 @@
|
||||||
LIBTORCH_VERSION="${LIBTORCH_VER:-1.10.0}"
|
LIBTORCH_VERSION="${LIBTORCH_VER:-1.10.0}"
|
||||||
CUDA_VERSION="${CUDA_VER:-11.1}"
|
CUDA_VERSION="${CUDA_VER:-11.1}"
|
||||||
|
|
||||||
if [[ -z "${CUDA_VERSION}"=="cpu" ]]; then
|
if [ "${CUDA_VERSION}"=="cpu" ]; then
|
||||||
CU_VERSION="cpu"
|
CU_VERSION="cpu"
|
||||||
else
|
else
|
||||||
CU_VERSION="cu${CUDA_VERSION//./}"
|
CU_VERSION="cu${CUDA_VERSION//./}"
|
||||||
|
|
Loading…
Reference in New Issue
Block a user