|
|
@@ -48,6 +48,9 @@ sync() {
|
|
|
install() {
|
|
|
bold "pytorch install"
|
|
|
case "$(uname)" in
|
|
|
+ "Linux")
|
|
|
+ [ -n "$(which cmake)" ] || sudo apt install -y cmake
|
|
|
+ ;;
|
|
|
"Darwin")
|
|
|
brew list automake > /dev/null 2>&1 || brew install automake > /dev/null
|
|
|
brew list cmake > /dev/null 2>&1 || brew install cmake > /dev/null
|
|
|
@@ -55,8 +58,8 @@ install() {
|
|
|
brew list glog > /dev/null 2>&1 || brew install glog > /dev/null
|
|
|
;;
|
|
|
esac
|
|
|
- [ -n "$(which virtualenv)" ] || ${python} -m pip install --force-reinstall --user --quiet virtualenv
|
|
|
- [ -d "${virtualenv}" ] || virtualenv --quiet -p ${python} ${virtualenv}
|
|
|
+ [ -n "$(python3 -m pip list --disable-pip-version-check | grep -w virtualenv)" ] || ${python} -m pip install --force-reinstall --user --quiet virtualenv
|
|
|
+ [ -d "${virtualenv}" ] || ${python} -m virtualenv --quiet -p ${python} ${virtualenv}
|
|
|
source ${virtualenv}/bin/activate
|
|
|
${python} -m pip install --quiet future leveldb numpy protobuf pydot python-gflags pyyaml scikit-image setuptools six hypothesis typing tqdm
|
|
|
${python} -m pip install --quiet ${third_party}/pytorch
|