溫馨提示×
您好，登錄后才能下訂單哦！
點(diǎn)擊 登錄注冊(cè) 即表示同意《億速云用戶服務(wù)條款》
您好，登錄后才能下訂單哦！
這篇文章主要介紹“pytorch-gpu安裝失敗怎么解決”，在日常操作中，相信很多人在pytorch-gpu安裝失敗怎么解決問(wèn)題上存在疑惑，小編查閱了各式資料，整理出簡(jiǎn)單好用的操作方法，希望對(duì)大家解答”pytorch-gpu安裝失敗怎么解決”的疑惑有所幫助！接下來(lái)，請(qǐng)跟著小編一起來(lái)學(xué)習(xí)吧！
ubuntu18 系統(tǒng)上已經(jīng)配置好gpu加速環(huán)境
安裝pytorch-gpu 后出現(xiàn)以下問(wèn)題：
import torch x = torch.Tensor(2,3) x_cuda = x.cuda()
x可以正常，當(dāng)定義x_cuda 后，命令行卡住，不再繼續(xù)輸出
幾周后終于找到了原因：
當(dāng)初pytorch-gpu為1.0版本，然而配置的cuda和cudnn版本較低，不支持高版本的pytorch,后來(lái)選擇安裝 0.4.1版本的pytorch，解決此問(wèn)題。
當(dāng)前環(huán)境是 torch-0.4.1-cp36 +384顯卡驅(qū)動(dòng)+cuda8.0+cudnn6.0。
幾多艱辛，得以開(kāi)顏。
補(bǔ)充：conda安裝pytorch-gpu版本下載出錯(cuò)的問(wèn)題解決
換了清華的源之后，官網(wǎng)的命令是下面的：
conda install pytorch=1.1.0 torchvision=0.3.0 cudatoolkit=9.0 -c pytorch
但是經(jīng)常會(huì)報(bào)HTTP error的錯(cuò)誤，將-c pytorch刪除即可
conda install pytorch=1.1.0 torchvision=0.3.0 cudatoolkit=9.0
到此，關(guān)于“pytorch-gpu安裝失敗怎么解決”的學(xué)習(xí)就結(jié)束了，希望能夠解決大家的疑惑。理論與實(shí)踐的搭配能更好的幫助大家學(xué)習(xí)，快去試試吧！若想繼續(xù)學(xué)習(xí)更多相關(guān)知識(shí)，請(qǐng)繼續(xù)關(guān)注億速云網(wǎng)站，小編會(huì)繼續(xù)努力為大家?guī)?lái)更多實(shí)用的文章！
免責(zé)聲明：本站發(fā)布的內(nèi)容（圖片、視頻和文字）以原創(chuàng)、轉(zhuǎn)載和分享為主，文章觀點(diǎn)不代表本網(wǎng)站立場(chǎng)，如果涉及侵權(quán)請(qǐng)聯(lián)系站長(zhǎng)郵箱：is@yisu.com進(jìn)行舉報(bào)，并提供相關(guān)證據(jù)，一經(jīng)查實(shí)，將立刻刪除涉嫌侵權(quán)內(nèi)容。
億速云公眾號(hào)
手機(jī)網(wǎng)站二維碼
