Skip to content

Commit

Permalink
Create requirements-gpu
Browse files Browse the repository at this point in the history
  • Loading branch information
angrave authored Feb 6, 2025
1 parent f1da56b commit 3c12621
Showing 1 changed file with 21 additions and 0 deletions.
21 changes: 21 additions & 0 deletions requirements-gpu
Original file line number Diff line number Diff line change
@@ -0,0 +1,21 @@
detectron2
lmdeploy==0.6.2.post1

nvidia-cublas-cu12==12.1.3.1
nvidia-cuda-cupti-cu12==12.1.105
nvidia-cuda-nvrtc-cu12==12.1.105
nvidia-cuda-runtime-cu12==12.1.105
nvidia-cudnn-cu12==8.9.2.26
nvidia-cufft-cu12==11.0.2.54
nvidia-curand-cu12==10.3.2.106
nvidia-cusolver-cu12==11.4.5.107
nvidia-cusparse-cu12==12.1.0.106
nvidia-nccl-cu12==2.20.5
nvidia-nvjitlink-cu12==12.6.77
nvidia-nvtx-cu12==12.1.105

# These didn't work on MAC OSX.
torch==2.3.1
torchtext==0.18.0
torchvision==0.18.1
triton==2.3.1

0 comments on commit 3c12621

Please sign in to comment.