RuntimeError: nvrtc: error: invalid value for --gpu-architecture (-arch)

[clear1111]

I'm encountering a CUDA architecture compatibility error when running the structure prediction step with RoseTTAFold. My CUDA version is 12, here's the specific error: RuntimeError: nvrtc: error: invalid value for --gpu-architecture (-arch)

I've set up the environment according to the instructions, but I'm stuck at this step. Could you please help me troubleshoot this issue? Thank you so much for your assistance! 🙏

[rclune]

It might be a versioning issue with some of the dependencies, namely PyTorch.

Perhaps the solution here may help: conda install pytorch==1.13.1 torchvision==0.14.1 torchaudio==0.13.1 pytorch-cuda=11.7 -c pytorch -c nvidia