From 0c027822a48440be59f9b4889831af7879ecf277 Mon Sep 17 00:00:00 2001 From: Jordan Gong Date: Tue, 12 Jan 2021 15:39:18 +0800 Subject: Update XLA requirements --- requirements.txt | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) (limited to 'requirements.txt') 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 -- cgit v1.2.3