Просмотр исходного кода

PyTorch script using pip install

Lutz Roeder 7 лет назад
Родитель
Сommit
19f6b6d22f
1 измененных файлов с 1 добавлено и 3 удалено
  1. 1 3
      tools/pytorch

+ 1 - 3
tools/pytorch

@@ -63,11 +63,9 @@ install() {
     [ -n "$(which virtualenv)" ] || ${python} -m pip install --force-reinstall --user --quiet virtualenv
     [ -d "${virtualenv}" ] || virtualenv --quiet -p ${python} ${virtualenv}
     source ${virtualenv}/bin/activate
-    pushd "${third_party}/pytorch" > /dev/null
     ${python} -m pip install --quiet future leveldb numpy protobuf pydot python-gflags pyyaml scikit-image setuptools six hypothesis typing tqdm
-    ${python} setup.py install
+    ${python} -m pip install --quiet ${third_party}/pytorch
     ${python} -m pip install --quiet torchvision
-    popd > /dev/null
     deactivate
 }