Files
tinygrad/requirements.txt
Omar Ben Brahim d0c3912517 fix for invalid GPU error caused by (test/test_net_speed.py::TestConvSpeed::test_mnist) when testing in CI (#31)
* install the pytorch cpu only version

* get rid of torch gpu version

* test passed seems to get rid of invalid gpu error

* added the libs to requirements.txt

* lol
2020-10-29 17:45:16 -07:00

9 lines
157 B
Plaintext

numpy
# util deps
requests
# testing deps
--find-links https://download.pytorch.org/whl/torch_stable.html
torch==1.6.0+cpu
torchvision==0.7.0+cpu
pytest
tqdm