simonduerr commited on
Commit
1f9981d
1 Parent(s): abb97b4

Update requirements_gpu.txt

Browse files
Files changed (1) hide show
  1. requirements_gpu.txt +22 -0
requirements_gpu.txt CHANGED
@@ -0,0 +1,22 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ transformers==4.19.2
2
+ absl-py==0.13.0
3
+ biopython==1.79
4
+ chex==0.0.7
5
+ dm-haiku==0.0.5
6
+ dm-tree==0.1.6
7
+ docker==5.0.0
8
+ immutabledict==2.0.0
9
+ jax[cuda]<0.3.0
10
+ jaxlib==0.1.76
11
+ ml-collections==0.1.0
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
18
+ GPUtil
19
+ numba
20
+ ray
21
+ protobuf<4
22
+ -f https://storage.googleapis.com/jax-releases/jax_releases.html