pytorch model.cuda()花費(fèi)時(shí)間很長(zhǎng)的解決
更新時(shí)間：2021年06月01日 10:22:40 作者：勵(lì)志學(xué)好數(shù)據(jù)結(jié)構(gòu)
這篇文章主要介紹了pytorch model.cuda()花費(fèi)時(shí)間很長(zhǎng)的解決方案，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。如有錯(cuò)誤或未考慮完全的地方，望不吝賜教
解決方法之一：
如果pytorch在進(jìn)行model.cuda(）操作需要花費(fèi)的時(shí)間很長(zhǎng)，長(zhǎng)到你懷疑GPU的速度了，那就是不正常的。
如果你用的pytorch版本是0.3.0，升級(jí)到0.3.1就好了！
.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版本的安裝包，
方法二：
conda install pytorch torchvision cuda80 -c soumith
這里可以根據(jù)自己cuda版本確定。查看cuda版本，在python命令行里面輸入：
import torch torch.version.cuda
即可。
以上為個(gè)人經(jīng)驗(yàn)，希望能給大家一個(gè)參考，也希望大家多多支持腳本之家。
您可能感興趣的文章:
相關(guān)文章
這篇文章主要給大家介紹了關(guān)于scrapy利用selenium爬取豆瓣閱讀的相關(guān)資料，文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧2020-09-09
這篇文章主要給大家介紹了關(guān)于python模擬練習(xí)題的相關(guān)資料,文中涉及質(zhì)因數(shù)分解、開(kāi)根變換、立方體拼接、日期計(jì)算、按位異或、停車(chē)場(chǎng)收費(fèi)、整數(shù)操作、減法運(yùn)算、相鄰數(shù)之和、以及最長(zhǎng)勾子序列的尋找,每題都有具體的輸入輸出示例和代碼計(jì)算過(guò)程,需要的朋友可以參考下2024-11-11
python3中bytes數(shù)據(jù)類(lèi)型的具體使用
bytes類(lèi)型是python3引入的,本文就來(lái)介紹一下python3中bytes數(shù)據(jù)類(lèi)型的具體使用,文中通過(guò)示例代碼介紹的非常詳細(xì),對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧2024-12-12
又是一年一度的圣誕節(jié)快到了,提到圣誕節(jié),就不得不提圣誕樹(shù),所以本文我們將使用Python繪制一棵圣誕樹(shù),文中有詳細(xì)的代碼講解,具有一定的參考價(jià)值,需要的朋友可以參考下2023-12-12
python機(jī)器學(xué)習(xí)之貝葉斯分類(lèi)
這篇文章主要為大家詳細(xì)介紹了python機(jī)器學(xué)習(xí)之貝葉斯分類(lèi)的相關(guān)資料，具有一定的參考價(jià)值，感興趣的小伙伴們可以參考一下2018-03-03
Python 比較文本相似性的方法(difflib,Levenshtein)
今天小編就為大家分享一篇Python 比較文本相似性的方法(difflib,Levenshtein)，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧2018-10-10
在PyTorch中,我們經(jīng)常需要處理多維數(shù)據(jù),如張量tensors,這篇文章主要介紹了一些Pytorch常用的維度轉(zhuǎn)換方法,文中通過(guò)代碼介紹的非常詳細(xì),需要的朋友可以參考下2025-03-03
python編寫(xiě)softmax函數(shù)、交叉熵函數(shù)實(shí)例
這篇文章主要介紹了python編寫(xiě)softmax函數(shù)、交叉熵函數(shù)實(shí)例，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧2020-06-06
Python爬蟲(chóng)實(shí)現(xiàn)爬取下載網(wǎng)站數(shù)據(jù)的幾種方法示例
這篇文章主要為大家介紹了Python爬蟲(chóng)實(shí)現(xiàn)爬取下載網(wǎng)站數(shù)據(jù)的幾種方法示例詳解，有需要的朋友可以借鑒參考下，希望能夠有所幫助，祝大家多多進(jìn)步，早日升職加薪2023-11-11
Python實(shí)現(xiàn)的tab文件操作類(lèi)分享
這篇文章主要介紹了Python實(shí)現(xiàn)的tab文件操作類(lèi)分享,本文直接給出實(shí)現(xiàn)代碼,需要的朋友可以參考下2014-11-11