pytorch-gpu安裝失敗怎么辦

這篇文章將為大家詳細(xì)講解有關(guān)pytorch-gpu安裝失敗怎么辦，小編覺得挺實(shí)用的，因此分享給大家做個(gè)參考，希望大家閱讀完這篇文章后可以有所收獲。

ubuntu18 系統(tǒng)上已經(jīng)配置好gpu加速環(huán)境，

安裝pytorch-gpu 后出現(xià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,后來選擇安裝 0.4.1版本的pytorch，解決此問題。

當(dāng)前環(huán)境是 torch-0.4.1-cp36 +384顯卡驅(qū)動(dòng)+cuda8.0+cudnn6.0。

幾多艱辛，得以開顏。

補(bǔ)充：conda安裝pytorch-gpu版本下載出錯(cuò)的問題解決

換了清華的源之后，官網(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

pytorch的優(yōu)點(diǎn)

1.PyTorch是相當(dāng)簡潔且高效快速的框架；2.設(shè)計(jì)追求最少的封裝；3.設(shè)計(jì)符合人類思維，它讓用戶盡可能地專注于實(shí)現(xiàn)自己的想法；4.與google的Tensorflow類似，F(xiàn)AIR的支持足以確保PyTorch獲得持續(xù)的開發(fā)更新；5.PyTorch作者親自維護(hù)的論壇 供用戶交流和求教問題6.入門簡單

關(guān)于“pytorch-gpu安裝失敗怎么辦”這篇文章就分享到這里了，希望以上內(nèi)容可以對(duì)大家有一定的幫助，使各位可以學(xué)到更多知識(shí)，如果覺得文章不錯(cuò)，請(qǐng)把它分享出去讓更多的人看到。