pytorch model.cuda()花費時間很長的解決
更新時間：2021年06月01日 10:22:40 作者：勵志學好數據結構
這篇文章主要介紹了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
后面這個可以自己到官網上找自己對應的python和cuda版本的安裝包，
方法二：
conda install pytorch torchvision cuda80 -c soumith
這里可以根據自己cuda版本確定。查看cuda版本，在python命令行里面輸入：
import torch torch.version.cuda
即可。
以上為個人經驗，希望能給大家一個參考，也希望大家多多支持腳本之家。
相關文章
這篇文章主要介紹了Pycharm 安裝 idea VIM的教程，本文通過圖文并茂的形式給大家介紹的非常詳細，具有一定的參考借鑒價值，需要的朋友可以參考下2020-02-02
python 中關于pycharm選擇運行環(huán)境的問題
這篇文章主要介紹了python 中關于pycharm選擇運行環(huán)境的相關知識，本文給大家介紹的非常詳細，對大家的學習或工作具有一定的參考借鑒價值，需要的朋友可以參考下2020-10-10
Python基于win32ui模塊創(chuàng)建彈出式菜單示例
這篇文章主要介紹了Python基于win32ui模塊創(chuàng)建彈出式菜單,結合實例形式分析了Python使用win32ui模塊創(chuàng)建彈出式菜單的具體步驟與相關操作技巧,并附帶說明了win32ui模塊的安裝命令,需要的朋友可以參考下2018-05-05
閑來無事，看到自己有很多網站的賬戶密碼，有些網站可能打開一兩次也就忘記了，下一次在輸入賬戶密碼就想不起來，這樣很容易丟失賬號。所以本文就來用Python和tkinter寫一個本地密碼管理器吧2023-05-05
這篇文章主要給大家介紹在?cpython?當中一些比較花里胡哨的魔術方法，以幫助我們自己實現比較花哨的功能，當然這其中也包含一些也非常實用的魔術方法，需要的可以參考下2023-05-05
今天小編就為大家分享一篇pytorch forward兩個參數實例，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2020-01-01
這篇文章主要介紹了Python使用設計模式中的責任鏈模式與迭代器模式的示例,責任鏈模式與迭代器模式都可以被看作為行為型的設計模式,需要的朋友可以參考下2016-03-03
Python使用virtualenv創(chuàng)建虛擬環(huán)境的詳細步驟
這篇文章主要介紹了Python使用virtualenv創(chuàng)建虛擬環(huán)境的詳細步驟,本文分步驟給大家介紹的非常詳細,感興趣的朋友跟隨小編一起看看吧2024-05-05
Python中輸入若干整數以逗號間隔實現統計每個整數出現次數
這篇文章主要介紹了Python中輸入若干整數以逗號間隔實現統計每個整數出現次數的相關資料,需要的小伙伴可以參考一下，希望對你有所幫助2022-04-04
這篇文章主要為大家詳細介紹了python實現彈窗祝福效果，具有一定的參考價值，感興趣的小伙伴們可以參考一下2019-04-04