simonduerr commited on
Commit
0f661de
·
1 Parent(s): 549e82e

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +1 -1
requirements.txt CHANGED
@@ -12,6 +12,6 @@ ml-collections==0.1.0
12
  numpy==1.19.5
13
  pandas==1.3.4
14
  scipy==1.7.0
15
- tensorflow-cpu==2.5.0
16
  torch
17
  plotly
 
12
  numpy==1.19.5
13
  pandas==1.3.4
14
  scipy==1.7.0
15
+ tensorflow-gpu==2.5.0
16
  torch
17
  plotly