最近在安裝pytorch時極為惱火,明明電腦有GPU且pytorch已經裝好,但是torch.cuda.is_available()一直是返回false。在網上搜集了一堆解決辦法,最終摸索了兩天后解決了這個問題。為避免下次遇到此問題,特此做個解決方法記錄。
當出現torch.cuda.is_available()返回false的情況時解決辦法
- 首先在自己創建的用於安裝pytorch的虛擬環境中,輸入
conda list
查看從官網下載的pytorch是CPU版本的還是GPU版本的。 - 若發現自己下載的pytorch為CPU版本的,則將此pytorch卸載,去官網使用PIP安裝方式安裝新的pytorch,記住千萬,千萬,千萬不要用conda的安裝方式,因為它下載的一直都是CPU版本的。
- 若自己下載的確實為GPU版本的,但是仍舊出現torch.cuda.is_available()返回false的情況,則可能是版本不匹配的問題。