pytorch無(wú)法使用GPU問(wèn)題的解決方法
本教程介紹了如何解決pytorch 無(wú)法使用GPU 的問(wèn)題，
問(wèn)題描述：
之前我電腦已經(jīng)安裝好CUDA 10.1 的環(huán)境，驅(qū)動(dòng)也是安裝好的，安裝了tensorflow-gpu ，并且可以正常使用tensorflow gpu 版來(lái)訓(xùn)練。今天有同學(xué)問(wèn)我一些pytorch 的安裝問(wèn)題，我之前沒(méi)有安裝過(guò)，所以實(shí)踐了一下，安裝過(guò)程很順利，找到官方網(wǎng)站，找到安裝的地方，復(fù)制命令，使用anconda 進(jìn)行了安裝：
安裝完成之后，我迫不及待的測(cè)試了一下是否安裝成功和能否使用GPU：
import torch print(torch.cuda.is_available())
結(jié)果返回的結(jié)果是False，也就是不可以使用GPU版本的pytorch，這讓人很煩。
問(wèn)題解決思路
根據(jù)以往tensorflow安裝時(shí)候的問(wèn)題解決思路，我猜測(cè)可能是cuda版本和pytorch 的版本對(duì)不上，我仔細(xì)對(duì)照了半天，發(fā)現(xiàn)不是這里的問(wèn)題。所以問(wèn)題可能出在cuda 和 顯卡驅(qū)動(dòng)上面。
我輸入命令：nvidia-smi，正常的話會(huì)輸出顯卡驅(qū)動(dòng)的相關(guān)信息，但是這里報(bào)錯(cuò)了，報(bào)錯(cuò)信息為：
NVIDIA-SMI has failed because it couldn’t communicate with the NVIDIA driver
顯卡驅(qū)動(dòng)出問(wèn)題了？ 那重裝一下驅(qū)動(dòng)應(yīng)該就可以解決問(wèn)題，但是在ubuntu上安裝驅(qū)動(dòng)有點(diǎn)繁瑣，所以我又測(cè)試了一下tensorflow-gpu 版本是否可以使用，輸入一下代碼：
import tensorflow as tf
發(fā)現(xiàn)竟然沒(méi)有報(bào)錯(cuò)，陷入僵局了，這一句話沒(méi)報(bào)錯(cuò)驅(qū)動(dòng)應(yīng)該沒(méi)問(wèn)題啊，我開(kāi)始查找問(wèn)題的答案，百度搜索了一下這個(gè)報(bào)錯(cuò)：NVIDIA-SMI has failed because it couldn’t communicate with the NVIDIA driver
最終得出的結(jié)論是就是顯卡驅(qū)動(dòng)壞了，我參考網(wǎng)上的解決辦法，執(zhí)行了以下命令，成功的解決了問(wèn)題。
首先安裝 dkms
sudo apt-get install dkms
然后查看自己驅(qū)動(dòng)的版本
ls /usr/local
輸出結(jié)果如下：
執(zhí)行以下命令：
sudo dkms install -m nvidia -v 418.43
注意命令后的版本要和上圖中的版本一致！
驗(yàn)證是否生效
nvidia-smi 命令執(zhí)行正常了。
驗(yàn)證pytorch：
問(wèn)題成功解決了
后記
剛剛提到了一個(gè)奇怪的現(xiàn)象，顯卡驅(qū)動(dòng)不正常，tensorflow-gpu 竟然可以成功的import，但是后來(lái)經(jīng)過(guò)測(cè)試發(fā)現(xiàn)，雖然沒(méi)有報(bào)錯(cuò)，但是訓(xùn)練過(guò)程使用了CPU，沒(méi)有使用GPU，運(yùn)行的時(shí)候會(huì)報(bào)錯(cuò)說(shuō) GPU驅(qū)動(dòng)沒(méi)有運(yùn)行之類的錯(cuò)誤，然后使用CPU進(jìn)行了訓(xùn)練（直觀感覺(jué)就是慢了很多）。
為什么會(huì)出現(xiàn)顯卡驅(qū)動(dòng)壞了的情況？猜測(cè)是系統(tǒng)更新造成的，因?yàn)橹拔沂褂胻ensorflow-gpu的時(shí)候是正常的，并且沒(méi)有做任何的改動(dòng)，所以很有可能是更新造成的損壞。
以上就是pytorch無(wú)法使用GPU問(wèn)題的解決方法的詳細(xì)內(nèi)容，更多關(guān)于pytorch無(wú)法使用GPU的資料請(qǐng)關(guān)注腳本之家其它相關(guān)文章！
相關(guān)文章
正則表達(dá)式是用來(lái)匹配與查找字符串的,從網(wǎng)上爬取數(shù)據(jù)自然或多或少會(huì)用到正則表達(dá)式,python的正則表達(dá)式要先引入re模塊,這篇文章主要給大家介紹了關(guān)于Python正則表達(dá)式re模塊的相關(guān)資料,需要的朋友可以參考下2022-07-07
Django 實(shí)現(xiàn)對(duì)已存在的model進(jìn)行更改
這篇文章主要介紹了Django 實(shí)現(xiàn)對(duì)已存在的model進(jìn)行更改，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧2020-03-03
Python實(shí)現(xiàn)讀取機(jī)器硬件信息的方法示例
這篇文章主要介紹了Python實(shí)現(xiàn)讀取機(jī)器硬件信息的方法,涉及Python針對(duì)計(jì)算機(jī)注冊(cè)表、操作系統(tǒng)、處理器、網(wǎng)絡(luò)等常見(jiàn)硬件信息讀取操作相關(guān)實(shí)現(xiàn)技巧,需要的朋友可以參考下2018-06-06
Python 圖形界面框架TkInter之在源碼中找pack方法
這篇文章主要介紹了Python 圖形界面框架TkInter之在源碼中找pack方法，根據(jù)不同的需求，選擇對(duì)應(yīng)的布局方式，每個(gè)控件都可以使用pack作為布局管理，下面我們就來(lái)看看從源碼中找pack（）在哪里,需要的朋友可以參考下2022-03-03
圣誕節(jié)快到了，今天小編通過(guò)代碼畫(huà)顆圣誕樹(shù)，主要通過(guò)t.pensize(10) 修改畫(huà)筆大小,本文通過(guò)實(shí)例代碼給大家介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或工作具有一定的參考借鑒價(jià)值，需要的朋友參考下吧2021-12-12
Python人工智能之路 之PyAudio 實(shí)現(xiàn)錄音 自動(dòng)化交互實(shí)現(xiàn)問(wèn)答
關(guān)于音頻, PyAudio 這個(gè)庫(kù), 可以實(shí)現(xiàn)開(kāi)啟麥克風(fēng)錄音, 可以播放音頻文件等等。文章介紹了如何使用Python第三方庫(kù)PyAudio進(jìn)行麥克風(fēng)錄音然后自動(dòng)播放已經(jīng)合成的語(yǔ)音實(shí)現(xiàn)語(yǔ)音交互回答,需要的朋友可以參考下2019-08-08
本文只是簡(jiǎn)單的對(duì)h5py庫(kù)的基本創(chuàng)建文件，數(shù)據(jù)集和讀取數(shù)據(jù)的方式進(jìn)行介紹，文中通過(guò)示例代碼介紹的非常詳細(xì)，具有一定的參考價(jià)值，感興趣的小伙伴們可以參考一下2022-03-03
今天小編就為大家分享一篇python中ASCII碼和字符的轉(zhuǎn)換方法，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧2018-07-07
Python使用translate庫(kù)實(shí)現(xiàn)多語(yǔ)種互相翻譯
文章介紹Python的translate庫(kù),支持40+語(yǔ)言互譯,可指定全稱或代碼進(jìn)行翻譯,能處理文本文件,自動(dòng)分割超500字符行并逐段翻譯,結(jié)果保存至新文件,需先通過(guò)pip安裝,用戶需提供文件路徑并檢查存在性,翻譯成功后顯示結(jié)果文件路徑2025-07-07
這篇文章主要為大家詳細(xì)介紹了pygame游戲之旅的第5篇，教大家如何在游戲中添加顯示文字，具有一定的參考價(jià)值，感興趣的小伙伴們可以參考一下2018-11-11