建站
咨詢
售后
建站咨詢
新聞中心
這里有您想知道的互聯(lián)網(wǎng)營(yíng)銷(xiāo)解決方案
pytorch-gpu安裝失敗解決方案
這篇文章主要介紹了解決pytorch-gpu 安裝失敗的記錄，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。如有錯(cuò)誤或未考慮完全的地方，望不吝賜教
網(wǎng)站建設(shè)哪家好，找創(chuàng)新互聯(lián)！專(zhuān)注于網(wǎng)頁(yè)設(shè)計(jì)、網(wǎng)站建設(shè)、微信開(kāi)發(fā)、微信小程序定制開(kāi)發(fā)、集團(tuán)企業(yè)網(wǎng)站建設(shè)等服務(wù)項(xiàng)目。為回饋新老客戶創(chuàng)新互聯(lián)還提供了市中免費(fèi)建站歡迎大家使用！
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
文章題目：pytorch-gpu安裝失敗解決方案
文章起源：http://m.fisionsoft.com.cn/article/dpheepe.html
