建站
咨詢
售后
建站咨詢
新聞中心
這里有您想知道的互聯網營銷解決方案
詳解pytorch-gpu安裝失敗具體解決方法
pytorch-gpu 在安裝過程中可以會出現很多錯誤導致安裝失敗，本篇文章為大家詳細講解一下在安裝pytorch-gpu過程中會遇到的問題以及解決方法，有需要的小伙伴可以參考一下。
創(chuàng)新互聯公司是專業(yè)的玉環(huán)網站建設公司，玉環(huán)接單;提供成都做網站、成都網站設計,網頁設計,網站設計,建網站,PHP網站建設等專業(yè)做網站服務;采用PHP框架,可快速的進行玉環(huán)網站開發(fā)網頁制作和功能擴展;專業(yè)做搜索引擎喜愛的網站,專業(yè)的做網站團隊,希望更多企業(yè)前來合作!
ubuntu18 系統(tǒng)上已經配置好gpu加速環(huán)境
安裝pytorch-gpu 后出現以下問題：
import torch
x = torch.Tensor(2,3)
x_cuda = x.cuda()
x可以正常，當定義x_cuda 后，命令行卡住，不再繼續(xù)輸出
幾周后終于找到了原因：
當初pytorch-gpu為1.0版本，然而配置的cuda和cudnn版本較低，不支持高版本的pytorch,后來選擇安裝 0.4.1版本的pytorch，解決此問題。
當前環(huán)境是 torch-0.4.1-cp36 +384顯卡驅動+cuda8.0+cudnn6.0。
幾多艱辛，得以開顏。
補充：conda安裝pytorch-gpu版本下載出錯的問題解決
換了清華的源之后，官網的命令是下面的：
conda install pytorch=1.1.0 torchvision=0.3.0 cudatoolkit=9.0 -c pytorch
但是經常會報HTTP error的錯誤，將-c pytorch刪除即可
conda install pytorch=1.1.0 torchvision=0.3.0 cudatoolkit=9.0
分享題目：詳解pytorch-gpu安裝失敗具體解決方法
文章URL：http://m.fisionsoft.com.cn/article/dhooeje.html
