Skip to content

Commit

Permalink
Update requirements for onnxruntime-gpu
Browse files Browse the repository at this point in the history
Needed for compatibility with CUDA 12.
  • Loading branch information
b-fission committed Oct 28, 2024
1 parent de14fac commit ea8de51
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 2 deletions.
2 changes: 1 addition & 1 deletion requirements_linux.txt
Original file line number Diff line number Diff line change
Expand Up @@ -8,6 +8,6 @@ xformers==0.0.28.post2
bitsandbytes==0.44.0
tensorboard==2.15.2
tensorflow==2.15.0.post1
onnxruntime-gpu==1.17.1
onnxruntime-gpu==1.19.2

-r requirements.txt
2 changes: 1 addition & 1 deletion requirements_windows.txt
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
bitsandbytes==0.44.0
tensorboard
tensorflow>=2.16.1
onnxruntime-gpu==1.17.1
onnxruntime-gpu==1.19.2

-r requirements.txt

0 comments on commit ea8de51

Please sign in to comment.