如何解決.cuda()載入用時很長

大雄45發表於2021-06-14
導讀 這篇文章主要介紹瞭如何解決.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 行裡面輸入：

如何解決.cuda()載入用時很長如何解決.cuda()載入用時很長

import torch
torch.version.cuda

即可。

補充：關於win10環境下在使用torch呼叫cuda（）失敗的原因

看了無數篇部落格都是牛頭不對馬嘴，忙活了半天才搞清楚

如果大家在用torch的過程中看見這樣的錯誤。。。。淡定，不是你的問題，也不是程式碼問題。。。是破win10的原因

如何解決.cuda()載入用時很長如何解決.cuda()載入用時很長

只需要在程式碼前面加上

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

就可以正常訓練了。。。心累啊

原文來自：  https://www.linuxprobe.com/time-consuming-cuda.html


來自 “ ITPUB部落格 ” ，連結：http://blog.itpub.net/69955379/viewspace-2776649/，如需轉載，請註明出處，否則將追究法律責任。

相關文章