software/wendelin: Add pytorch.

e5783fc6 · Yusei Tahara · 7b02a24f · e5783fc6
Commit e5783fc6 authored Oct 17, 2017 by Yusei Tahara
Show whitespace changes
Inline Side-by-side

Showing with 5 additions and 0 deletions

software/wendelin/software-kerastensorflow.cfg software/wendelin/software-kerastensorflow.cfg +5 -0

No files found.
--- a/software/wendelin/software-kerastensorflow.cfg
+++ b/software/wendelin/software-kerastensorflow.cfg
@@ -2,10 +2,12 @@
 extends =
  ../../component/keras/buildout.cfg
  ../../component/chainer/buildout.cfg
+  ../../component/pytorch/buildout.cfg
  software.cfg
 parts +=
  keras-egg
  chainer-egg
+  pytorch-egg
 [eggs]
 eggs +=
@@ -19,6 +21,9 @@ eggs +=
  filelock
  nose
  ${chainer:egg}
+  ${pytorch-build-install-egg:egg}
+  ${python-cocoapi-build-install-egg:egg}
+  torchvision
 [cuda]
 # If you use cuda, please adjust paramters for your environment