diff options
| author | Jordan Gong <jordan.gong@protonmail.com> | 2021-01-12 15:39:18 +0800 |
|---|---|---|
| committer | Jordan Gong <jordan.gong@protonmail.com> | 2021-01-12 15:39:18 +0800 |
| commit | 0c027822a48440be59f9b4889831af7879ecf277 (patch) | |
| tree | a5815d88d0eaf0f19265b5ec794f0999c2f0b633 | |
| parent | 61f79339fd0168bc200117daf0459764b93d0acb (diff) | |
Update XLA requirements
| -rw-r--r-- | requirements.txt | 7 |
1 files changed, 4 insertions, 3 deletions
diff --git a/requirements.txt b/requirements.txt index deaa0d4..e110c5d 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,6 +1,7 @@ -torch~=1.7.1 -torchvision~=0.8.0a0+ecf4e9c +torch~=1.7.1+cpu +torch-xla~=1.7 +torchvision~=0.8.2+cpu numpy~=1.19.4 tqdm~=4.56.0 Pillow~=8.1.0 -scikit-learn~=0.23.2
\ No newline at end of file +scikit-learn~=0.24.0
\ No newline at end of file |