主頁 > 知識庫 > 解決pytorch-gpu 安裝失敗的記錄

解決pytorch-gpu 安裝失敗的記錄

熱門標簽:開封語音外呼系統(tǒng)代理商 應電話機器人打電話違法嗎 電銷機器人的風險 開封自動外呼系統(tǒng)怎么收費 手機網(wǎng)頁嵌入地圖標注位置 天津電話機器人公司 400電話辦理哪種 地圖標注線上如何操作 河北防封卡電銷卡

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

安裝pytorch-gpu 后出現(xiàn)以下問題:

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顯卡驅(qū)動+cuda8.0+cudnn6.0。

幾多艱辛,得以開顏。

補充:conda安裝pytorch-gpu版本下載出錯的問題解決

換了清華的源之后,官網(wǎng)的命令是下面的:

conda install pytorch=1.1.0 torchvision=0.3.0 cudatoolkit=9.0 -c pytorch

但是經(jīng)常會報HTTP error的錯誤,將-c pytorch刪除即可

conda install pytorch=1.1.0 torchvision=0.3.0 cudatoolkit=9.0

以上為個人經(jīng)驗,希望能給大家一個參考,也希望大家多多支持腳本之家。

您可能感興趣的文章:
  • PyTorch-GPU加速實例
  • Pytorch 搭建分類回歸神經(jīng)網(wǎng)絡(luò)并用GPU進行加速的例子
  • pytorch 兩個GPU同時訓練的解決方案
  • Pytorch 如何查看、釋放已關(guān)閉程序占用的GPU資源
  • Linux環(huán)境下GPU版本的pytorch安裝
  • 如何判斷pytorch是否支持GPU加速

標簽:駐馬店 六盤水 江蘇 蘭州 成都 常州 宿遷 山東

巨人網(wǎng)絡(luò)通訊聲明:本文標題《解決pytorch-gpu 安裝失敗的記錄》,本文關(guān)鍵詞  解決,pytorch-gpu,安裝,失,;如發(fā)現(xiàn)本文內(nèi)容存在版權(quán)問題,煩請?zhí)峁┫嚓P(guān)信息告之我們,我們將及時溝通與處理。本站內(nèi)容系統(tǒng)采集于網(wǎng)絡(luò),涉及言論、版權(quán)與本站無關(guān)。
  • 相關(guān)文章
  • 下面列出與本文章《解決pytorch-gpu 安裝失敗的記錄》相關(guān)的同類信息!
  • 本頁收集關(guān)于解決pytorch-gpu 安裝失敗的記錄的相關(guān)信息資訊供網(wǎng)民參考!
  • 推薦文章