algorithmiaio / langpacks

Standardized builder and runners for Algorithmia algorithms
https://algorithmia.com/
MIT License
16 stars 9 forks source link

Added support for PyTorch 1.6.x on CPU #167

Closed aslisabanci closed 4 years ago

aslisabanci commented 4 years ago

Hey James! I also added support for PyTorch 1.6.x on CPU so that we can merge this one together with TF2.3.0 on CPU to legit. I built the package and tested it on deepPurple and the test is successful. Can you please review - thank you!!

./tools/environment_validator.py -b ubuntu:16.04 -g python3 -s python37 -d pytorch-cpu-1.6.x -t dependency -n pytorch-cpu-1.6.x -c 1
curl localhost:9999 -H 'Content-Type: application/json' -d '{ "matrix_a": [[0, 1], [1, 0]], "matrix_b": [[25, 25], [11, 11]] }'