python中cuda()加載用時(shí)很長(zhǎng)的解決方法

發(fā)布時(shí)間：2022-02-23 14:35:49 來(lái)源：億速云閱讀：144 作者：小新欄目：開(kāi)發(fā)技術(shù)

這篇文章將為大家詳細(xì)講解有關(guān)python中cuda()加載用時(shí)很長(zhǎng)的解決方法，小編覺(jué)得挺實(shí)用的，因此分享給大家做個(gè)參考，希望大家閱讀完這篇文章后可以有所收獲。

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

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的原因

只需要在代碼前面加上

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

就可以正常訓(xùn)練了。

關(guān)于“python中cuda()加載用時(shí)很長(zhǎng)的解決方法”這篇文章就分享到這里了，希望以上內(nèi)容可以對(duì)大家有一定的幫助，使各位可以學(xué)到更多知識(shí)，如果覺(jué)得文章不錯(cuò)，請(qǐng)把它分享出去讓更多的人看到。

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

猜你喜歡