如何解決.cuda()加載用時(shí)很長(zhǎng)的問題
方法一:
pip install --upgrade --force-reinstall http://download.pytorch.org/whl/cu80/torch-0.2.0.post3-cp27-cp27mu-manylinux1_x86_64.whl
后面這個(gè)可以自己到官網(wǎng)上找自己對(duì)應(yīng)的python和cuda版本的安裝包,官網(wǎng)為鏈接:
方法二:
conda install pytorch torchvision cuda80 -c soumith
這里可以根據(jù)自己cuda版本確定。查看cuda版本,在python命令行里面輸入:
import torch torch.version.cuda
即可。
補(bǔ)充:關(guān)于win10環(huán)境下在使用torch調(diào)用cuda()失敗的原因
看了無數(shù)篇博客都是牛頭不對(duì)馬嘴,忙活了半天才搞清楚
如果大家在用torch的過程中看見這樣的錯(cuò)誤。。。。淡定,不是你的問題,也不是代碼問題。。。是破win10的原因
只需要在代碼前面加上
torch.cuda.current_device() torch.cuda._initialized = True
就可以正常訓(xùn)練了。。。心累啊
相關(guān)文章
python讀取中文路徑時(shí)出錯(cuò)(2種解決方案)
這篇文章主要介紹了python讀取中文路徑時(shí)出錯(cuò)的解決方案,具有很好的參考價(jià)值,希望對(duì)大家有所幫助。一起跟隨小編過來看看吧2021-03-03Python實(shí)現(xiàn)確認(rèn)字符串是否包含指定字符串的實(shí)例
下面小編就為大家分享一篇Python實(shí)現(xiàn)確認(rèn)字符串是否包含指定字符串的實(shí)例,具有很好的參考價(jià)值,希望對(duì)大家有所幫助。一起跟隨小編過來看看吧2018-05-05python實(shí)現(xiàn)excel讀寫數(shù)據(jù)
這篇文章主要為大家詳細(xì)介紹了python操作EXCEL讀數(shù)據(jù)、寫數(shù)據(jù)的實(shí)例源碼,文中示例代碼介紹的非常詳細(xì),具有一定的參考價(jià)值,感興趣的小伙伴們可以參考一下2018-04-04django 實(shí)現(xiàn)將本地圖片存入數(shù)據(jù)庫(kù),并能顯示在web上的示例
今天小編就為大家分享一篇django 實(shí)現(xiàn)將本地圖片存入數(shù)據(jù)庫(kù),并能顯示在web上的示例,具有很好的參考價(jià)值,希望對(duì)大家有所幫助。一起跟隨小編過來看看吧2019-08-08利用Pandas讀取表格行數(shù)據(jù)判斷是否相同的方法
這篇文章主要給大家介紹了關(guān)于利用Pandas讀取表格行數(shù)據(jù)判斷是否相同的相關(guān)資料,文中通過示例代碼介紹的非常詳細(xì),對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2021-03-03python機(jī)器學(xué)習(xí)之神經(jīng)網(wǎng)絡(luò)(一)
這篇文章主要為大家詳細(xì)介紹了python機(jī)器學(xué)習(xí)之神經(jīng)網(wǎng)絡(luò)第一篇,具有一定的參考價(jià)值,感興趣的小伙伴們可以參考一下2017-12-12