Instructions for Blackwell architecture?

[Raize]

Due diligence

• I have done my due diligence in trying to find the answer myself.

Topic

The PyTorch implementation

Question

Getting this error when attempting to run the server on Blackwell architecture:

UserWarning:
NVIDIA RTX PRO 6000 Blackwell Server Edition with CUDA capability sm_120 is not compatible with the current PyTorch installation.
The current PyTorch install supports CUDA capabilities sm_50 sm_60 sm_70 sm_75 sm_80 sm_86 sm_90.
If you want to use the NVIDIA RTX PRO 6000 Blackwell Server Edition GPU with PyTorch, please check the instructions at https://pytorch.org/get-started/locally/

I wasn't able to find anything about Moshi not working with Blackwell, though I suppose that might be possible?

[Raize]

So the follow up is the following DID fix the issue:
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu128

I suppose the question about if the instructions should be modified to include running this manually still kind of stands, though.

[rajarshiroy-nvidia]

Thank you! This will help others. Added this to the README.