Commit e5783fc6 authored by Yusei Tahara's avatar Yusei Tahara

software/wendelin: Add pytorch.

parent 7b02a24f
......@@ -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
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment