File size: 443 Bytes
70ca20a |
1 2 3 4 5 6 7 8 9 10 11 |
# Common dependencies -r requirements-common.txt # Dependencies for NVIDIA GPUs ray >= 2.9 nvidia-ml-py # for pynvml package torch == 2.4.0 # These must be updated alongside torch torchvision == 0.19 # Required for phi3v processor. See https://github.com/pytorch/vision?tab=readme-ov-file#installation for corresponding version xformers == 0.0.27.post2; platform_system == 'Linux' and platform_machine == 'x86_64' # Requires PyTorch 2.4.0 |