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)文章
python 中關(guān)于pycharm選擇運行環(huán)境的問題
這篇文章主要介紹了python 中關(guān)于pycharm選擇運行環(huán)境的相關(guān)知識,本文給大家介紹的非常詳細(xì),對大家的學(xué)習(xí)或工作具有一定的參考借鑒價值,需要的朋友可以參考下2020-10-10Python基于win32ui模塊創(chuàng)建彈出式菜單示例
這篇文章主要介紹了Python基于win32ui模塊創(chuàng)建彈出式菜單,結(jié)合實例形式分析了Python使用win32ui模塊創(chuàng)建彈出式菜單的具體步驟與相關(guān)操作技巧,并附帶說明了win32ui模塊的安裝命令,需要的朋友可以參考下2018-05-05Python使用設(shè)計模式中的責(zé)任鏈模式與迭代器模式的示例
這篇文章主要介紹了Python使用設(shè)計模式中的責(zé)任鏈模式與迭代器模式的示例,責(zé)任鏈模式與迭代器模式都可以被看作為行為型的設(shè)計模式,需要的朋友可以參考下2016-03-03Python使用virtualenv創(chuàng)建虛擬環(huán)境的詳細(xì)步驟
這篇文章主要介紹了Python使用virtualenv創(chuàng)建虛擬環(huán)境的詳細(xì)步驟,本文分步驟給大家介紹的非常詳細(xì),感興趣的朋友跟隨小編一起看看吧2024-05-05Python中輸入若干整數(shù)以逗號間隔實現(xiàn)統(tǒng)計每個整數(shù)出現(xiàn)次數(shù)
這篇文章主要介紹了Python中輸入若干整數(shù)以逗號間隔實現(xiàn)統(tǒng)計每個整數(shù)出現(xiàn)次數(shù)的相關(guān)資料,需要的小伙伴可以參考一下,希望對你有所幫助2022-04-04