安裝pytorch時,torch.cuda.is_available()返回false的解決辦法


    最近在安裝pytorch時極為惱火,明明電腦有GPU且pytorch已經裝好,但是torch.cuda.is_available()一直是返回false。在網上搜集了一堆解決辦法,最終摸索了兩天后解決了這個問題。為避免下次遇到此問題,特此做個解決方法記錄。
當出現torch.cuda.is_available()返回false的情況時解決辦法

  1. 首先在自己創建的用於安裝pytorch的虛擬環境中,輸入conda list查看從官網下載的pytorch是CPU版本的還是GPU版本的。
  2. 若發現自己下載的pytorch為CPU版本的,則將此pytorch卸載,去官網使用PIP安裝方式安裝新的pytorch記住千萬,千萬,千萬不要用conda的安裝方式,因為它下載的一直都是CPU版本的
  3. 若自己下載的確實為GPU版本的,但是仍舊出現torch.cuda.is_available()返回false的情況,則可能是版本不匹配的問題。


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM