安裝pytorch報錯torch.cuda.is_available()=false問題的解決過程
問題介紹
在安裝torch之后,命令行(Anaconda Powershell Prompt)運行這三行代碼:
python # python import torch torch.cuda.is_available()
返回結果始終為False。
出錯原因
原因有多個,可以參考文章最后的鏈接【1】
他的很清晰,如果按我的沒有解決可以看一下。
主要就是以下兩個:
- CUDA、cudnn、torch版本不對應。(解決方法參考鏈接【1】)
- 一個坑:是通過清華源下載的!
檢查是不是清華源下載導致的問題:
conda list
看到Pytorch的Build是pyxxx_cpu_0,就說明下的pytorch是cpu版本的。這樣就會導致輸出False。
導致的原因可能是清華源沒有對應cuda版本的pytorch,所以會裝cpu版的。參考鏈接【2】
我的問題就是通過清華源下載導致的。下面是解決方案
解決方案
上官網下載:https://pytorch.org/get-started/previous-versions/
找到你的CUDA版本支持的pytorch(我的是CUDA10.1)
然后復制對應的pip語句
——重點!復制pip語句,而不是下圖這種conda語句,因為你已經切換了清華源
然后在命令行(或Anaconda Prompt中運行)
記得不要科學上網,否則可能會報錯 ValueError: check_hostname requires server_hostname
等待,安裝就可以啦
要是下載速度很慢,可以參考鏈接【1】。
參考內容:
【1】torch.cuda.is_available()返回false——解決辦法
【2】完美解決torch.cuda.is_available()一直返回False的玄學方法
總結
到此這篇關于安裝pytorch報錯torch.cuda.is_available()=false問題解決的文章就介紹到這了,更多相關pytorch報錯torch.cuda.is_available()=false內容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關文章希望大家以后多多支持腳本之家!
相關文章
python處理emoji表情(兩個函數解決兩者之間的聯(lián)系)
這篇文章主要介紹了python處理emoji表情,主要通過兩個函數解決兩者之間的聯(lián)系,本文通過實例代碼給大家介紹的非常完美,對python emoji表情的相關知識感興趣的朋友一起看看吧2021-05-05Python3.9環(huán)境搭建RobotFramework的詳細過程
Robot Framework是一個基于Python的,可擴展的關鍵字驅動的測試自動化框架,用于端到端驗收測試和驗收測試驅動開發(fā)(ATDD),這篇文章主要介紹了Python3.9環(huán)境搭建RobotFramework的詳細過程,需要的朋友可以參考下2023-01-01Python greenlet和gevent使用代碼示例解析
這篇文章主要介紹了Python greenlet和gevent使用代碼示例解析,文中通過示例代碼介紹的非常詳細,對大家的學習或者工作具有一定的參考學習價值,需要的朋友可以參考下2020-04-04