PyTorch 2.7.0/CUDA 12.8 update?

[RLWOHIO]

Hi @potinglai , PyTorch 2.7.0 just came out. Any chance BioRex can be updated to this latest version?

Thanks!

[ptlai]

Hi @RLWOHIO,

I just tested BioREx with PyTorch 2.7.0 + CUDA 12.8 on WSL 2 and everything works as expected—no code changes are needed. The only update is the install command. Please replace:

pip install torch==2.6.0 torchvision==0.21.0 torchaudio==2.6.0 --index-url https://download.pytorch.org/whl/cu126

with:

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu128

per the PyTorch installation guide.

If you run into any issues, let me know the exact error message and a brief description of your environment.

Thanks!

[RLWOHIO]

Hi @ptlai , This is great to know. Thanks so much for the quick help!