最新国产好看的视频,伊人天堂AV在线,国产Aaaaaa视频,蜜臀视频在线观看一区,人妻av色图,密臀久久久精品影片,青青视频免费观看毛片,久草在线观看视,国产三级精品色情在线

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

 更新時間:2021年05月24日 14:57:40   作者:燁然_  
這篇文章主要介紹了解決pytorch-gpu 安裝失敗的記錄,具有很好的參考價值,希望對大家有所幫助。如有錯誤或未考慮完全的地方,望不吝賜教

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)常會報HTTP error的錯誤,將-c pytorch刪除即可

conda install pytorch=1.1.0 torchvision=0.3.0 cudatoolkit=9.0

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

相關(guān)文章

最新評論

泗洪县| 长武县| 九寨沟县| 黔西县| 永清县| 达拉特旗| 宁陵县| 武安市| 靖西县| 华池县| 湛江市| 克什克腾旗| 桂林市| 碌曲县| 介休市| 金堂县| 通州区| 资兴市| 宜兰市| 镇平县| 光山县| 镇沅| 靖远县| 浪卡子县| 泾川县| 西乌珠穆沁旗| 板桥市| 寿阳县| 浦江县| 诸暨市| 陵川县| 如东县| 棋牌| 新竹县| 南丹县| 永兴县| 安图县| 法库县| 石屏县| 泽普县| 和硕县|