CUDA

CUDA

CUDA VISIBLE DEVICE http://www.reibang.com/p/22c0f8ec9a3e

CUDA_VISIBLE_DEVICES=2,3 python my_script.py # Uses GPUs 2 and 3.
代碼里設(shè)置可見(jiàn)顯卡

if args.gpu == 'None':
    config.set_cuda(False)
else:
    try:
        os.environ['CUDA_VISIBLE_DEVICES'] = str(args.gpu)
    except IndexError:
        config.set_cuda(False)

如果要周期性的輸出顯卡的使用情況大年,可以用watch指令實(shí)現(xiàn):
watch -n 10 nvidia-smi

升級(jí)CUDA版本

看NVIDIA驅(qū)動(dòng)版本
sudo dpkg --list | grep nvidia-*nvidia-smi差不多波势,以前者為準(zhǔn)场靴,然后到 https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html不同版本cuda對(duì)應(yīng)的NVIDIA驅(qū)動(dòng)版本 (mamicode.com) 找對(duì)應(yīng)的 CUDA 版本买鸽;
檢查linux版本 sudo dpkg --list | grep nvidia-*
選擇CUDA版本https://developer.nvidia.com/cuda-11.1.0-download-archive?target_os=Linux&target_arch=x86_64&target_distro=WSLUbuntu&target_version=20 或者 https://developer.nvidia.com/cuda-downloads?target_os=Linux

CUDA版本檢測(cè)https://zhuanlan.zhihu.com/p/48641682

當(dāng)CUDA9安裝完成后,需要進(jìn)行檢測(cè):

  1. cat /usr/local/cuda/version.txt #這種方法不太準(zhǔn)镜雨,得到9.0.176

  2. nvcc --version #這種方法準(zhǔn)乙嘀。如果nvcc沒(méi)有安裝当叭,則sudo apt install nvidia-cuda-toolkit。檢查出來(lái)是7.5.17

3)通過(guò)代碼來(lái)查:

cudaDriverGetVersion(&driver_version); #獲取cuda版本职祷。得到9.0

cudaRuntimeGetVersion(&runtime_version); #獲取cuda運(yùn)行時(shí)版本氏涩。得到7.5

4)檢測(cè)顯卡型號(hào)及NVIDIA驅(qū)動(dòng)版本

查看GPU型號(hào) lspci | grep -i nvidia 得到M4000

查看NVIDIA驅(qū)動(dòng)版本 sudo dpkg --list | grep nvidia-* 得到384.130

根據(jù)不同版本cuda對(duì)應(yīng)的NVIDIA驅(qū)動(dòng)版本(不同版本cuda對(duì)應(yīng)的NVIDIA驅(qū)動(dòng)版本 - LearnFromNow - 博客園),這個(gè)驅(qū)動(dòng)應(yīng)該對(duì)應(yīng)cuda9.0

5)進(jìn)一步查是否環(huán)境變量的問(wèn)題,參見(jiàn) Ubuntu16.04下安裝多版本cuda和cudnn - tunhuzhuang1836的博客 - CSDN博客

在~/.bashrc中查看有無(wú)類似的語(yǔ)句有梆,如有改成下面的樣子是尖,沒(méi)有則增加:

export PATH="$PATH:/usr/local/cuda/bin"

export LD_LIBRARY_PATH="$LD_LIBRARY_PATH:/usr/local/cuda/lib64"

export LIBRARY_PATH="$LIBRARY_PATH:/usr/local/cuda/lib64"

然后繼續(xù)nvcc --version還是得到7.5。

6)檢查nvcc

which nvcc泥耀,發(fā)現(xiàn)是/usr/bin/nvcc饺汹,而實(shí)際上nvcc是應(yīng)該在/usr/local/cuda/bin/nvcc。

所以痰催,sudo gedit /usr/bin/nvcc兜辞,把里面的內(nèi)容"exec /usr/lib/nvidia-cuda-toolkit/bin/nvcc" 改成"exec /usr/local/cuda/bin/nvcc"

然后繼續(xù)nvcc --version這次得到9.0

DGL安裝

https://www.dgl.ai/pages/start.html

安裝 base 包

  • 更新pip
    python -m pip install --upgrade pip
  • 安裝 sklearn
    pip install -U scikit-learn
  • 1、使用Pip更新Pytorch和torchvision

列舉pip當(dāng)前可以更新的所有安裝包

pip list --outdated --format=legacy

更新pytorch和torchvision安裝包

pip install --upgrade pytorch torchvision

  • 2夸溶、使用conda更新Pytorch和torchvision

建議將其添加soumith為您的Anaconda(或Miniconda)的源服務(wù)器

conda config --add channels soumith

更新pytorch和torchvision安裝包

conda update pytorch torchvision
檢查
import torch
print(torch.version)
torch.cuda.is_available()返回 False
torch.cuda.is_available()返回False,但nvidia-smi正常

?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請(qǐng)聯(lián)系作者
  • 序言:七十年代末逸吵,一起剝皮案震驚了整個(gè)濱河市,隨后出現(xiàn)的幾起案子缝裁,更是在濱河造成了極大的恐慌扫皱,老刑警劉巖,帶你破解...
    沈念sama閱讀 216,402評(píng)論 6 499
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件捷绑,死亡現(xiàn)場(chǎng)離奇詭異韩脑,居然都是意外死亡,警方通過(guò)查閱死者的電腦和手機(jī)胎食,發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 92,377評(píng)論 3 392
  • 文/潘曉璐 我一進(jìn)店門(mén)扰才,熙熙樓的掌柜王于貴愁眉苦臉地迎上來(lái),“玉大人厕怜,你說(shuō)我怎么就攤上這事衩匣±僮埽” “怎么了?”我有些...
    開(kāi)封第一講書(shū)人閱讀 162,483評(píng)論 0 353
  • 文/不壞的土叔 我叫張陵琅捏,是天一觀的道長(zhǎng)生百。 經(jīng)常有香客問(wèn)我,道長(zhǎng)柄延,這世上最難降的妖魔是什么蚀浆? 我笑而不...
    開(kāi)封第一講書(shū)人閱讀 58,165評(píng)論 1 292
  • 正文 為了忘掉前任,我火速辦了婚禮搜吧,結(jié)果婚禮上市俊,老公的妹妹穿的比我還像新娘。我一直安慰自己滤奈,他們只是感情好摆昧,可當(dāng)我...
    茶點(diǎn)故事閱讀 67,176評(píng)論 6 388
  • 文/花漫 我一把揭開(kāi)白布。 她就那樣靜靜地躺著蜒程,像睡著了一般绅你。 火紅的嫁衣襯著肌膚如雪。 梳的紋絲不亂的頭發(fā)上昭躺,一...
    開(kāi)封第一講書(shū)人閱讀 51,146評(píng)論 1 297
  • 那天忌锯,我揣著相機(jī)與錄音,去河邊找鬼领炫。 笑死偶垮,一個(gè)胖子當(dāng)著我的面吹牛,可吹牛的內(nèi)容都是我干的帝洪。 我是一名探鬼主播针史,決...
    沈念sama閱讀 40,032評(píng)論 3 417
  • 文/蒼蘭香墨 我猛地睜開(kāi)眼,長(zhǎng)吁一口氣:“原來(lái)是場(chǎng)噩夢(mèng)啊……” “哼碟狞!你這毒婦竟也來(lái)了啄枕?” 一聲冷哼從身側(cè)響起,我...
    開(kāi)封第一講書(shū)人閱讀 38,896評(píng)論 0 274
  • 序言:老撾萬(wàn)榮一對(duì)情侶失蹤族沃,失蹤者是張志新(化名)和其女友劉穎频祝,沒(méi)想到半個(gè)月后,有當(dāng)?shù)厝嗽跇?shù)林里發(fā)現(xiàn)了一具尸體脆淹,經(jīng)...
    沈念sama閱讀 45,311評(píng)論 1 310
  • 正文 獨(dú)居荒郊野嶺守林人離奇死亡常空,尸身上長(zhǎng)有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點(diǎn)故事閱讀 37,536評(píng)論 2 332
  • 正文 我和宋清朗相戀三年,在試婚紗的時(shí)候發(fā)現(xiàn)自己被綠了盖溺。 大學(xué)時(shí)的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片漓糙。...
    茶點(diǎn)故事閱讀 39,696評(píng)論 1 348
  • 序言:一個(gè)原本活蹦亂跳的男人離奇死亡,死狀恐怖烘嘱,靈堂內(nèi)的尸體忽然破棺而出昆禽,到底是詐尸還是另有隱情蝗蛙,我是刑警寧澤,帶...
    沈念sama閱讀 35,413評(píng)論 5 343
  • 正文 年R本政府宣布醉鳖,位于F島的核電站捡硅,受9級(jí)特大地震影響,放射性物質(zhì)發(fā)生泄漏盗棵。R本人自食惡果不足惜壮韭,卻給世界環(huán)境...
    茶點(diǎn)故事閱讀 41,008評(píng)論 3 325
  • 文/蒙蒙 一、第九天 我趴在偏房一處隱蔽的房頂上張望纹因。 院中可真熱鬧喷屋,春花似錦、人聲如沸瞭恰。這莊子的主人今日做“春日...
    開(kāi)封第一講書(shū)人閱讀 31,659評(píng)論 0 22
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽(yáng)寄疏。三九已至,卻和暖如春僵井,著一層夾襖步出監(jiān)牢的瞬間陕截,已是汗流浹背。 一陣腳步聲響...
    開(kāi)封第一講書(shū)人閱讀 32,815評(píng)論 1 269
  • 我被黑心中介騙來(lái)泰國(guó)打工批什, 沒(méi)想到剛下飛機(jī)就差點(diǎn)兒被人妖公主榨干…… 1. 我叫王不留农曲,地道東北人。 一個(gè)月前我還...
    沈念sama閱讀 47,698評(píng)論 2 368
  • 正文 我出身青樓驻债,卻偏偏與公主長(zhǎng)得像乳规,于是被迫代替她去往敵國(guó)和親。 傳聞我的和親對(duì)象是個(gè)殘疾皇子合呐,可洞房花燭夜當(dāng)晚...
    茶點(diǎn)故事閱讀 44,592評(píng)論 2 353

推薦閱讀更多精彩內(nèi)容