GPU Not Supported

[xiaoa008]

Hi, when I use rfd3 in a PC with a NVIDIA GeForce RTX 5060 GPU, it reports "Failed to get GPU information from pynvml: Not Supported. Using default values for NVIDIA GeForce RTX 5060 Laptop GPU.
GPU information: NVIDIA RTX A6000, 8, 6, 45, 84, 300, 2100"

Will foundry add 5060 GPU support ?

[rclune]

A previous user had a similar issue with this system: #105
I believe the solution was to update PyTorch.

[xiaoa008]

A previous user had a similar issue with this system: #105 I believe the solution was to update PyTorch.

Updating Pytorch seems to work, but it raises another error "Error while loading libcue_ops.so: undefined symbol: cublasGemmGroupedBatchedEx, version libcublas.so.12"

[rclune]

Do you know what version of CUDA is running on your system?

[xiaoa008]

Do you know what version of CUDA is running on your system?

the latest CUDA version (13.1).

[rclune]

That would explain the issue then. It is likely that some of your dependencies (like PyTorch) are looking for a CUDA 12.x installation, which would generate files like libcublas.12, while your system has CUDA 13.1, which would instead have the file libcublas.13.

If you have access to a CUDA 12.x installation or can install it, that would be one fix. If there is a PyTorch version for CUDA 13.x you can try switching to this, but you'll likely need to change other dependencies as well.