{"id":51675,"date":"2024-10-08T20:37:16","date_gmt":"2024-10-08T12:37:16","guid":{"rendered":"https:\/\/server.hk\/cnblog\/51675\/"},"modified":"2024-11-14T10:42:17","modified_gmt":"2024-11-14T02:42:17","slug":"ubuntu-18-04-%e5%ae%89%e8%a3%9d-gpu-cuda-cudnn-%e4%bb%a5%e5%8f%8a%e5%85%b6%e7%89%88%e6%9c%ac%e9%81%b8%e6%93%87","status":"publish","type":"post","link":"https:\/\/server.hk\/cnblog\/51675\/","title":{"rendered":"Ubuntu 18.04 \u5b89\u88dd GPU + CUDA + cuDNN \u4ee5\u53ca\u5176\u7248\u672c\u9078\u64c7"},"content":{"rendered":"<h1 id=\"ubuntu-18-04-%e5%ae%89%e8%a3%9d-gpu-cuda-cudnn-%e4%bb%a5%e5%8f%8a%e5%85%b6%e7%89%88%e6%9c%ac%e9%81%b8%e6%93%87-agxjYXuICD\">Ubuntu 18.04 \u5b89\u88dd GPU + CUDA + cuDNN \u4ee5\u53ca\u5176\u7248\u672c\u9078\u64c7<\/h1>\n<p>\u5728\u6df1\u5ea6\u5b78\u7fd2\u548c\u9ad8\u6027\u80fd\u8a08\u7b97\u7684\u9818\u57df\u4e2d\uff0cGPU\uff08\u5716\u5f62\u8655\u7406\u55ae\u5143\uff09\u5df2\u6210\u70ba\u4e0d\u53ef\u6216\u7f3a\u7684\u5de5\u5177\u3002\u5c0d\u65bc\u4f7f\u7528 Ubuntu 18.04 \u7684\u958b\u767c\u8005\u4f86\u8aaa\uff0c\u5b89\u88dd GPU \u9a45\u52d5\u7a0b\u5f0f\u3001CUDA \u548c cuDNN \u662f\u4e00\u500b\u91cd\u8981\u7684\u6b65\u9a5f\u3002\u672c\u6587\u5c07\u8a73\u7d30\u4ecb\u7d39\u5982\u4f55\u5728 Ubuntu 18.04 \u4e0a\u5b89\u88dd\u9019\u4e9b\u7d44\u4ef6\uff0c\u4e26\u63d0\u4f9b\u7248\u672c\u9078\u64c7\u7684\u5efa\u8b70\u3002<\/p>\n<h2 id=\"%e4%b8%80%e3%80%81%e6%ba%96%e5%82%99%e5%b7%a5%e4%bd%9c-agxjYXuICD\">\u4e00\u3001\u6e96\u5099\u5de5\u4f5c<\/h2>\n<p>\u5728\u958b\u59cb\u5b89\u88dd\u4e4b\u524d\uff0c\u8acb\u78ba\u4fdd\u60a8\u7684\u7cfb\u7d71\u5df2\u7d93\u66f4\u65b0\u5230\u6700\u65b0\u7248\u672c\u3002\u53ef\u4ee5\u4f7f\u7528\u4ee5\u4e0b\u547d\u4ee4\u9032\u884c\u66f4\u65b0\uff1a<\/p>\n<pre><code>sudo apt update\nsudo apt upgrade<\/code><\/pre>\n<p>\u63a5\u4e0b\u4f86\uff0c\u6aa2\u67e5\u60a8\u7684 GPU \u578b\u865f\uff0c\u78ba\u4fdd\u5b83\u652f\u6301 CUDA\u3002\u53ef\u4ee5\u4f7f\u7528\u4ee5\u4e0b\u547d\u4ee4\u4f86\u6aa2\u67e5\uff1a<\/p>\n<pre><code>lspci | grep -i nvidia<\/code><\/pre>\n<h2 id=\"%e4%ba%8c%e3%80%81%e5%ae%89%e8%a3%9d-nvidia-%e9%a9%85%e5%8b%95%e7%a8%8b%e5%bc%8f-agxjYXuICD\">\u4e8c\u3001\u5b89\u88dd NVIDIA \u9a45\u52d5\u7a0b\u5f0f<\/h2>\n<p>\u9996\u5148\uff0c\u60a8\u9700\u8981\u5b89\u88dd NVIDIA \u9a45\u52d5\u7a0b\u5f0f\u3002\u53ef\u4ee5\u901a\u904e\u4ee5\u4e0b\u6b65\u9a5f\u9032\u884c\u5b89\u88dd\uff1a<\/p>\n<ol>\n<li>\u6dfb\u52a0 NVIDIA PPA\uff1a<\/li>\n<pre><code>sudo add-apt-repository ppa:graphics-drivers\/ppa<\/code><\/pre>\n<li>\u66f4\u65b0\u5305\u5217\u8868\uff1a<\/li>\n<pre><code>sudo apt update<\/code><\/pre>\n<li>\u5b89\u88dd NVIDIA \u9a45\u52d5\u7a0b\u5f0f\uff1a<\/li>\n<pre><code>sudo apt install nvidia-driver-<\/code><\/pre>\n<\/ol>\n<p>\u8acb\u5c07  \u66ff\u63db\u70ba\u60a8\u6240\u9700\u7684\u9a45\u52d5\u7248\u672c\u865f\u3002\u5b89\u88dd\u5b8c\u6210\u5f8c\uff0c\u91cd\u555f\u7cfb\u7d71\u4ee5\u4f7f\u9a45\u52d5\u751f\u6548\u3002<\/p>\n<h2 id=\"%e4%b8%89%e3%80%81%e5%ae%89%e8%a3%9d-cuda-agxjYXuICD\">\u4e09\u3001\u5b89\u88dd CUDA<\/h2>\n<p>\u63a5\u4e0b\u4f86\uff0c\u60a8\u9700\u8981\u5b89\u88dd CUDA\u3002\u53ef\u4ee5\u5f9e NVIDIA \u7684\u5b98\u65b9\u7db2\u7ad9\u4e0b\u8f09 CUDA \u5b89\u88dd\u5305\u3002\u4ee5\u4e0b\u662f\u5b89\u88dd\u6b65\u9a5f\uff1a<\/p>\n<ol>\n<li>\u8a2a\u554f NVIDIA CUDA Toolkit \u7db2\u7ad9\uff0c\u9078\u64c7\u9069\u5408 Ubuntu 18.04 \u7684\u7248\u672c\u3002<\/li>\n<li>\u4e0b\u8f09 .deb \u6587\u4ef6\u4e26\u5b89\u88dd\uff1a<\/li>\n<pre><code>sudo dpkg -i cuda-repo-.deb\nsudo apt-key adv --fetch-keys https:\/\/developer.download.nvidia.com\/compute\/cuda\/repos\/ubuntu1804\/x86_64\/7fa2af80.pub\nsudo apt update\nsudo apt install cuda<\/code><\/pre>\n<\/ol>\n<p>\u5b89\u88dd\u5b8c\u6210\u5f8c\uff0c\u60a8\u9700\u8981\u5c07 CUDA \u7684\u8def\u5f91\u6dfb\u52a0\u5230\u74b0\u5883\u8b8a\u91cf\u4e2d\u3002\u53ef\u4ee5\u7de8\u8f2f ~\/.bashrc \u6587\u4ef6\uff0c\u6dfb\u52a0\u4ee5\u4e0b\u884c\uff1a<\/p>\n<pre><code>export PATH=\/usr\/local\/cuda\/bin:$PATH\nexport LD_LIBRARY_PATH=\/usr\/local\/cuda\/lib64:$LD_LIBRARY_PATH<\/code><\/pre>\n<p>\u7136\u5f8c\u904b\u884c\u4ee5\u4e0b\u547d\u4ee4\u4f7f\u66f4\u6539\u751f\u6548\uff1a<\/p>\n<pre><code>source ~\/.bashrc<\/code><\/pre>\n<h2 id=\"%e5%9b%9b%e3%80%81%e5%ae%89%e8%a3%9d-cudnn-agxjYXuICD\">\u56db\u3001\u5b89\u88dd cuDNN<\/h2>\n<p>cuDNN \u662f NVIDIA \u63d0\u4f9b\u7684\u6df1\u5ea6\u5b78\u7fd2\u52a0\u901f\u5eab\u3002\u8981\u5b89\u88dd cuDNN\uff0c\u60a8\u9700\u8981\u5f9e NVIDIA \u958b\u767c\u8005\u7db2\u7ad9\u4e0b\u8f09\u76f8\u61c9\u7248\u672c\u7684 cuDNN\u3002\u4ee5\u4e0b\u662f\u5b89\u88dd\u6b65\u9a5f\uff1a<\/p>\n<ol>\n<li>\u8a2a\u554f NVIDIA cuDNN \u7db2\u7ad9\uff0c\u9078\u64c7\u8207\u60a8\u5b89\u88dd\u7684 CUDA \u7248\u672c\u76f8\u5c0d\u61c9\u7684 cuDNN \u7248\u672c\u3002<\/li>\n<li>\u4e0b\u8f09 cuDNN \u7684 tar \u6587\u4ef6\u4e26\u89e3\u58d3\uff1a<\/li>\n<pre><code>tar -xzvf cudnn--linux-x64-v.tgz<\/code><\/pre>\n<li>\u5c07\u89e3\u58d3\u5f8c\u7684\u6587\u4ef6\u8907\u88fd\u5230 CUDA \u76ee\u9304\uff1a<\/li>\n<pre><code>sudo cp cuda\/include\/cudnn*.h \/usr\/local\/cuda\/include\nsudo cp cuda\/lib64\/libcudnn* \/usr\/local\/cuda\/lib64\nsudo chmod a+r \/usr\/local\/cuda\/include\/cudnn*.h \/usr\/local\/cuda\/lib64\/libcudnn*<\/code><\/pre>\n<\/ol>\n<h2 id=\"%e4%ba%94%e3%80%81%e7%89%88%e6%9c%ac%e9%81%b8%e6%93%87%e5%bb%ba%e8%ad%b0-agxjYXuICD\">\u4e94\u3001\u7248\u672c\u9078\u64c7\u5efa\u8b70<\/h2>\n<p>\u5728\u9078\u64c7 CUDA \u548c cuDNN \u7684\u7248\u672c\u6642\uff0c\u5efa\u8b70\u9075\u5faa\u4ee5\u4e0b\u539f\u5247\uff1a<\/p>\n<ul>\n<li>\u78ba\u4fdd CUDA \u548c cuDNN \u7684\u7248\u672c\u76f8\u4e92\u517c\u5bb9\u3002\u53ef\u4ee5\u53c3\u8003 NVIDIA \u7684\u5b98\u65b9\u6587\u6a94\u4f86\u78ba\u8a8d\u7248\u672c\u517c\u5bb9\u6027\u3002<\/li>\n<li>\u9078\u64c7\u7a69\u5b9a\u7248\u672c\u800c\u975e\u6e2c\u8a66\u7248\u672c\uff0c\u4ee5\u78ba\u4fdd\u7cfb\u7d71\u7684\u7a69\u5b9a\u6027\u548c\u6027\u80fd\u3002<\/li>\n<li>\u6839\u64da\u60a8\u7684\u6df1\u5ea6\u5b78\u7fd2\u6846\u67b6\uff08\u5982 TensorFlow \u6216 PyTorch\uff09\u4f86\u9078\u64c7\u5408\u9069\u7684\u7248\u672c\uff0c\u56e0\u70ba\u4e0d\u540c\u6846\u67b6\u5c0d CUDA \u548c cuDNN \u7684\u652f\u6301\u7248\u672c\u53ef\u80fd\u6709\u6240\u4e0d\u540c\u3002<\/li>\n<\/ul>\n<h2 id=\"%e7%b8%bd%e7%b5%90-agxjYXuICD\">\u7e3d\u7d50<\/h2>\n<p>\u5728 Ubuntu 18.04 \u4e0a\u5b89\u88dd GPU\u3001CUDA \u548c cuDNN \u662f\u4e00\u500b\u76f8\u5c0d\u76f4\u63a5\u7684\u904e\u7a0b\uff0c\u4f46\u9700\u8981\u6ce8\u610f\u7248\u672c\u7684\u517c\u5bb9\u6027\u548c\u6b63\u78ba\u7684\u5b89\u88dd\u6b65\u9a5f\u3002\u901a\u904e\u9075\u5faa\u4e0a\u8ff0\u6307\u5357\uff0c\u60a8\u53ef\u4ee5\u70ba\u6df1\u5ea6\u5b78\u7fd2\u548c\u9ad8\u6027\u80fd\u8a08\u7b97\u642d\u5efa\u4e00\u500b\u5f37\u5927\u7684\u74b0\u5883\u3002\u5982\u679c\u60a8\u9700\u8981\u66f4\u9ad8\u6548\u7684\u8a08\u7b97\u8cc7\u6e90\uff0c\u53ef\u4ee5\u8003\u616e\u4f7f\u7528\u6211\u5011\u7684<a href=\"https:\/\/server.hk\">\u9999\u6e2fVPS<\/a>\u670d\u52d9\uff0c\u70ba\u60a8\u7684\u9805\u76ee\u63d0\u4f9b\u7a69\u5b9a\u7684\u652f\u6301\u3002<\/p>\n","protected":false},"excerpt":{"rendered":"<p>Learn how to install GPU, CUDA, and cuDNN on Ubuntu 18.04, including tips on selecting the right versions for optimal performance.<\/p>\n","protected":false},"author":1,"featured_media":0,"comment_status":"closed","ping_status":"","sticky":false,"template":"","format":"standard","meta":{"footnotes":""},"categories":[4924],"tags":[],"class_list":["post-51675","post","type-post","status-publish","format-standard","hentry","category-setup-tutorials"],"_links":{"self":[{"href":"https:\/\/server.hk\/cnblog\/wp-json\/wp\/v2\/posts\/51675","targetHints":{"allow":["GET"]}}],"collection":[{"href":"https:\/\/server.hk\/cnblog\/wp-json\/wp\/v2\/posts"}],"about":[{"href":"https:\/\/server.hk\/cnblog\/wp-json\/wp\/v2\/types\/post"}],"author":[{"embeddable":true,"href":"https:\/\/server.hk\/cnblog\/wp-json\/wp\/v2\/users\/1"}],"replies":[{"embeddable":true,"href":"https:\/\/server.hk\/cnblog\/wp-json\/wp\/v2\/comments?post=51675"}],"version-history":[{"count":1,"href":"https:\/\/server.hk\/cnblog\/wp-json\/wp\/v2\/posts\/51675\/revisions"}],"predecessor-version":[{"id":51676,"href":"https:\/\/server.hk\/cnblog\/wp-json\/wp\/v2\/posts\/51675\/revisions\/51676"}],"wp:attachment":[{"href":"https:\/\/server.hk\/cnblog\/wp-json\/wp\/v2\/media?parent=51675"}],"wp:term":[{"taxonomy":"category","embeddable":true,"href":"https:\/\/server.hk\/cnblog\/wp-json\/wp\/v2\/categories?post=51675"},{"taxonomy":"post_tag","embeddable":true,"href":"https:\/\/server.hk\/cnblog\/wp-json\/wp\/v2\/tags?post=51675"}],"curies":[{"name":"wp","href":"https:\/\/api.w.org\/{rel}","templated":true}]}}