Commit e5783fc6 authored by Yusei Tahara's avatar Yusei Tahara

software/wendelin: Add pytorch.

parent 7b02a24f
...@@ -2,10 +2,12 @@ ...@@ -2,10 +2,12 @@
extends = extends =
../../component/keras/buildout.cfg ../../component/keras/buildout.cfg
../../component/chainer/buildout.cfg ../../component/chainer/buildout.cfg
../../component/pytorch/buildout.cfg
software.cfg software.cfg
parts += parts +=
keras-egg keras-egg
chainer-egg chainer-egg
pytorch-egg
[eggs] [eggs]
eggs += eggs +=
...@@ -19,6 +21,9 @@ eggs += ...@@ -19,6 +21,9 @@ eggs +=
filelock filelock
nose nose
${chainer:egg} ${chainer:egg}
${pytorch-build-install-egg:egg}
${python-cocoapi-build-install-egg:egg}
torchvision
[cuda] [cuda]
# If you use cuda, please adjust paramters for your environment # 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