Open billyphuse opened 1 year ago
方法1:將pytorch更改為對應的cuda版本即可解決(是虛擬環境的pytorch,不是本機的喔) 方法2:將本機cuda版本改為11.6(此方法本人沒有試過,但理論上應該可行) 希望有幫助到您
How to change cuda version ?
Found it ;-)
I edit setup.bat line 22: venv\Scripts\pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
And now it see my GPU
1.nvidia controll panel -> get cuda version 2.go visit pytorch official website -> get cuda pytorch download url 3.change 'setup.bat' line22 torch install url