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