怎么解決.cuda()加載用時(shí)很長(zhǎng)的問(wèn)題

發(fā)布時(shí)間：2021-05-25 13:54:56 來(lái)源：億速云閱讀：146 作者：小新欄目：開(kāi)發(fā)技術(shù)

這篇文章主要介紹怎么解決.cuda()加載用時(shí)很長(zhǎng)的問(wèn)題，文中介紹的非常詳細(xì)，具有一定的參考價(jià)值，感興趣的小伙伴們一定要看完！

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命令行里面輸入：

怎么解決.cuda()加載用時(shí)很長(zhǎng)的問(wèn)題

import torch
torch.version.cuda

即可。

補(bǔ)充：關(guān)于win10環(huán)境下在使用torch調(diào)用cuda（）失敗的原因

看了無(wú)數(shù)篇博客都是牛頭不對(duì)馬嘴，忙活了半天才搞清楚

如果大家在用torch的過(guò)程中看見(jiàn)這樣的錯(cuò)誤。。。。淡定，不是你的問(wèn)題，也不是代碼問(wèn)題。。。是破win10的原因

怎么解決.cuda()加載用時(shí)很長(zhǎng)的問(wèn)題

只需要在代碼前面加上

torch.cuda.current_device()
torch.cuda._initialized = True

以上是“怎么解決.cuda()加載用時(shí)很長(zhǎng)的問(wèn)題”這篇文章的所有內(nèi)容，感謝各位的閱讀！希望分享的內(nèi)容對(duì)大家有幫助，更多相關(guān)知識(shí)，歡迎關(guān)注億速云行業(yè)資訊頻道！

向AI問(wèn)一下細(xì)節(jié)

猜你喜歡