或許有小伙伴可能在剛開始學(xué)pytorch的時候也遇到過小編這樣的問題:在定義cuda的時候命令行卡住不再繼續(xù)輸出。這其實(shí)是pytorch-gpu安裝失敗造成的,那么pytorch-gpu安裝失敗怎么辦?如何解決?今天小編就將我解決的方法告訴大家。
關(guān)于配置
小編當(dāng)時用的是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ū)動+cuda8.0+cudnn6.0。
補(bǔ)充:conda安裝pytorch-gpu版本下載出錯的問題解決
換了清華的源之后,官網(wǎng)的命令是下面的:
conda install pytorch=1.1.0 torchvision=0.3.0 cudatoolkit=9.0 -c pytorch
但是經(jīng)常會報(bào)HTTP error的錯誤,將-c pytorch刪除即可
conda install pytorch=1.1.0 torchvision=0.3.0 cudatoolkit=9.0
小結(jié)
以上就是pytorch-gpu安裝失敗怎么辦的解決方式,也許不同的環(huán)境下出現(xiàn)的錯誤也不相同,但是當(dāng)遇到跟小編相同的問題的時候可以嘗試一下小編的方案,或許就解決了呢?以上為個人經(jīng)驗(yàn),希望能給大家一個參考,也希望大家多多支持W3Cschool。