溫馨提示×
您好，登錄后才能下訂單哦！
點擊 登錄注冊 即表示同意《億速云用戶服務條款》
您好，登錄后才能下訂單哦！
這篇文章將為大家詳細講解有關pytorch-gpu安裝失敗怎么辦，小編覺得挺實用的，因此分享給大家做個參考，希望大家閱讀完這篇文章后可以有所收獲。
ubuntu18 系統上已經配置好gpu加速環境，
import torch x = torch.Tensor(2,3) x_cuda = x.cuda()
x可以正常，當定義x_cuda 后，命令行卡住，不再繼續輸出
解決方法正在尋找中。。。找到后將記錄解決方案。
當初pytorch-gpu為1.0版本，然而配置的cuda和cudnn版本較低，不支持高版本的pytorch,后來選擇安裝 0.4.1版本的pytorch，解決此問題。
當前環境是 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
1.PyTorch是相當簡潔且高效快速的框架；2.設計追求最少的封裝；3.設計符合人類思維，它讓用戶盡可能地專注于實現自己的想法；4.與google的Tensorflow類似，FAIR的支持足以確保PyTorch獲得持續的開發更新；5.PyTorch作者親自維護的論壇 供用戶交流和求教問題6.入門簡單
關于“pytorch-gpu安裝失敗怎么辦”這篇文章就分享到這里了，希望以上內容可以對大家有一定的幫助，使各位可以學到更多知識，如果覺得文章不錯，請把它分享出去讓更多的人看到。
免責聲明：本站發布的內容（圖片、視頻和文字）以原創、轉載和分享為主，文章觀點不代表本網站立場，如果涉及侵權請聯系站長郵箱：is@yisu.com進行舉報，并提供相關證據，一經查實，將立刻刪除涉嫌侵權內容。
