解決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可以正常,當定義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)驗,希望能給大家一個參考,也希望大家多多支持腳本之家。
相關(guān)文章
Django web自定義通用權(quán)限控制實現(xiàn)方法
這篇文章主要介紹了Django web自定義通用權(quán)限控制實現(xiàn)方法,文中通過示例代碼介紹的非常詳細,對大家的學習或者工作具有一定的參考學習價值,需要的朋友可以參考下2020-11-11
Python實現(xiàn)復制文件從一個目錄到另外一個目錄
這篇文章主要為大家詳細介紹了如何使用Python實現(xiàn)復制文件從一個目錄到另外一個目錄,文中的示例代碼簡潔易懂,有需要的小伙伴可以參考一下2025-01-01
pycharm實現(xiàn)在虛擬環(huán)境中引入別人的項目
這篇文章主要介紹了pycharm實現(xiàn)在虛擬環(huán)境中引入別人的項目,具有很好的參考價值,希望對大家有所幫助。一起跟隨小編過來看看吧2020-03-03

