解決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)文章
Django web自定義通用權(quán)限控制實(shí)現(xiàn)方法
這篇文章主要介紹了Django web自定義通用權(quán)限控制實(shí)現(xiàn)方法,文中通過示例代碼介紹的非常詳細(xì),對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值,需要的朋友可以參考下2020-11-11
Python實(shí)現(xiàn)復(fù)制文件從一個目錄到另外一個目錄
這篇文章主要為大家詳細(xì)介紹了如何使用Python實(shí)現(xiàn)復(fù)制文件從一個目錄到另外一個目錄,文中的示例代碼簡潔易懂,有需要的小伙伴可以參考一下2025-01-01
pycharm實(shí)現(xiàn)在虛擬環(huán)境中引入別人的項目
這篇文章主要介紹了pycharm實(shí)現(xiàn)在虛擬環(huán)境中引入別人的項目,具有很好的參考價值,希望對大家有所幫助。一起跟隨小編過來看看吧2020-03-03
如何用python實(shí)現(xiàn)復(fù)制粘貼功能
這篇文章主要介紹了如何用python實(shí)現(xiàn)復(fù)制粘貼功能,幫助大家更好的理解和學(xué)習(xí)使用python,感興趣的朋友可以了解下2021-03-03
Python實(shí)現(xiàn)從PPT中導(dǎo)出高分辨率圖片
這篇文章主要為大家分享了一個實(shí)用腳本——如何利用Python實(shí)現(xiàn)從PPT中導(dǎo)出高分辨率(高 dpi)的圖片,文中的示例代碼講解詳細(xì),感興趣的可以了解一下2023-05-05
python 多線程實(shí)現(xiàn)多任務(wù)的方法示例
本文主要介紹了python 多線程實(shí)現(xiàn)多任務(wù)的方法示例,文中通過示例代碼介紹的非常詳細(xì),需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2021-07-07

