gpu acceleration

[nepenthes123]

How to use gpu acceleration, I cannot install the corresponding version of dependency on the windows system, and an error still occurs after installing the related dependency in ubuntu20.04. May I ask which version of the system I should run on

[Jackma-Q7]

pip install "torch==2.3.1+cu118" "torchvision==0.18.1+cu118" --index-url https://download.pytorch.org/whl/cu118