ubuntu18 系統(tǒng)上已經(jīng)配置好gpu加速環(huá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
以上為個(gè)人經(jīng)驗(yàn),希望能給大家一個(gè)參考,也希望大家多多支持腳本之家。
標(biāo)簽:駐馬店 六盤(pán)水 江蘇 蘭州 成都 常州 宿遷 山東
巨人網(wǎng)絡(luò)通訊聲明:本文標(biāo)題《解決pytorch-gpu 安裝失敗的記錄》,本文關(guān)鍵詞 解決,pytorch-gpu,安裝,失,;如發(fā)現(xiàn)本文內(nèi)容存在版權(quán)問(wèn)題,煩請(qǐng)?zhí)峁┫嚓P(guān)信息告之我們,我們將及時(shí)溝通與處理。本站內(nèi)容系統(tǒng)采集于網(wǎng)絡(luò),涉及言論、版權(quán)與本站無(wú)關(guān)。