{"id":51413,"date":"2024-10-08T18:24:14","date_gmt":"2024-10-08T10:24:14","guid":{"rendered":"https:\/\/server.hk\/cnblog\/51413\/"},"modified":"2024-11-14T10:43:21","modified_gmt":"2024-11-14T02:43:21","slug":"ubuntu-16-04-%e5%ae%89%e8%a3%9d-caffe-gpu-%e7%89%88","status":"publish","type":"post","link":"https:\/\/server.hk\/cnblog\/51413\/","title":{"rendered":"Ubuntu 16.04 \u5b89\u88dd Caffe GPU \u7248"},"content":{"rendered":"<h1 id=\"ubuntu-16-04-%e5%ae%89%e8%a3%9d-caffe-gpu-%e7%89%88-OhhAhQyDiK\">Ubuntu 16.04 \u5b89\u88dd Caffe GPU \u7248<\/h1>\n<p>\u5728\u6df1\u5ea6\u5b78\u7fd2\u7684\u9818\u57df\u4e2d\uff0cCaffe \u662f\u4e00\u500b\u975e\u5e38\u53d7\u6b61\u8fce\u7684\u6846\u67b6\uff0c\u7279\u5225\u662f\u5728\u5716\u50cf\u8655\u7406\u548c\u8a08\u7b97\u6a5f\u8996\u89ba\u65b9\u9762\u3002Caffe \u7684 GPU \u7248\u672c\u80fd\u5920\u5229\u7528 NVIDIA \u7684 CUDA \u6280\u8853\u4f86\u52a0\u901f\u8a08\u7b97\uff0c\u5f9e\u800c\u986f\u8457\u63d0\u9ad8\u6a21\u578b\u8a13\u7df4\u7684\u6548\u7387\u3002\u672c\u6587\u5c07\u4ecb\u7d39\u5982\u4f55\u5728 Ubuntu 16.04 \u4e0a\u5b89\u88dd Caffe \u7684 GPU \u7248\u672c\u3002<\/p>\n<h2 id=\"%e5%89%8d%e7%bd%ae%e6%a2%9d%e4%bb%b6-OhhAhQyDiK\">\u524d\u7f6e\u689d\u4ef6<\/h2>\n<p>\u5728\u958b\u59cb\u5b89\u88dd\u4e4b\u524d\uff0c\u8acb\u78ba\u4fdd\u60a8\u7684\u7cfb\u7d71\u6eff\u8db3\u4ee5\u4e0b\u8981\u6c42\uff1a<\/p>\n<ul>\n<li>Ubuntu 16.04 \u64cd\u4f5c\u7cfb\u7d71<\/li>\n<li>NVIDIA GPU\uff08\u652f\u6301 CUDA\uff09<\/li>\n<li>\u5df2\u5b89\u88dd NVIDIA \u9a45\u52d5\u7a0b\u5e8f<\/li>\n<li>\u5df2\u5b89\u88dd CUDA Toolkit<\/li>\n<li>\u5df2\u5b89\u88dd cuDNN<\/li>\n<\/ul>\n<h2 id=\"%e5%ae%89%e8%a3%9d%e6%ad%a5%e9%a9%9f-OhhAhQyDiK\">\u5b89\u88dd\u6b65\u9a5f<\/h2>\n<h3 id=\"1-%e6%9b%b4%e6%96%b0%e7%b3%bb%e7%b5%b1-OhhAhQyDiK\">1. \u66f4\u65b0\u7cfb\u7d71<\/h3>\n<pre><code>sudo apt-get update\nsudo apt-get upgrade<\/code><\/pre>\n<h3 id=\"2-%e5%ae%89%e8%a3%9d%e4%be%9d%e8%b3%b4%e9%a0%85-OhhAhQyDiK\">2. \u5b89\u88dd\u4f9d\u8cf4\u9805<\/h3>\n<p>\u5728\u5b89\u88dd Caffe \u4e4b\u524d\uff0c\u9700\u8981\u5b89\u88dd\u4e00\u4e9b\u5fc5\u8981\u7684\u4f9d\u8cf4\u9805\uff1a<\/p>\n<pre><code>sudo apt-get install build-essential cmake git libatlas-base-dev libboost-all-dev libhdf5-serial-dev protobuf-compiler libgoogle-glog-dev libgflags-dev libopencv-dev<\/code><\/pre>\n<h3 id=\"3-%e4%b8%8b%e8%bc%89-caffe-%e6%ba%90%e7%a2%bc-OhhAhQyDiK\">3. \u4e0b\u8f09 Caffe \u6e90\u78bc<\/h3>\n<p>\u4f7f\u7528 Git \u4e0b\u8f09 Caffe \u7684\u6e90\u78bc\uff1a<\/p>\n<pre><code>git clone https:\/\/github.com\/BVLC\/caffe.git\ncd caffe<\/code><\/pre>\n<h3 id=\"4-%e9%85%8d%e7%bd%ae-caffe-OhhAhQyDiK\">4. \u914d\u7f6e Caffe<\/h3>\n<p>\u5728 Caffe \u7684\u6839\u76ee\u9304\u4e2d\uff0c\u6709\u4e00\u500b\u540d\u70ba <code>Makefile.config.example<\/code> \u7684\u7bc4\u672c\u6587\u4ef6\u3002\u5c07\u5176\u8907\u88fd\u4e26\u91cd\u547d\u540d\u70ba <code>Makefile.config<\/code>\uff1a<\/p>\n<pre><code>cp Makefile.config.example Makefile.config<\/code><\/pre>\n<p>\u63a5\u4e0b\u4f86\uff0c\u4f7f\u7528\u6587\u672c\u7de8\u8f2f\u5668\u6253\u958b <code>Makefile.config<\/code>\uff0c\u4e26\u6839\u64da\u60a8\u7684\u7cfb\u7d71\u914d\u7f6e\u9032\u884c\u4fee\u6539\u3002\u78ba\u4fdd\u555f\u7528 GPU \u652f\u6301\uff1a<\/p>\n<pre><code>USE_CUDA := 1\nUSE_CUDNN := 1<\/code><\/pre>\n<h3 id=\"5-%e7%b7%a8%e8%ad%af-caffe-OhhAhQyDiK\">5. \u7de8\u8b6f Caffe<\/h3>\n<p>\u5728\u914d\u7f6e\u5b8c\u6210\u5f8c\uff0c\u4f7f\u7528\u4ee5\u4e0b\u547d\u4ee4\u7de8\u8b6f Caffe\uff1a<\/p>\n<pre><code>make -j8\nmake test -j8\nmake runtest -j8<\/code><\/pre>\n<h3 id=\"6-%e5%ae%89%e8%a3%9d-python-%e6%8e%a5%e5%8f%a3%ef%bc%88%e5%8f%af%e9%81%b8%ef%bc%89-OhhAhQyDiK\">6. \u5b89\u88dd Python \u63a5\u53e3\uff08\u53ef\u9078\uff09<\/h3>\n<p>\u5982\u679c\u60a8\u5e0c\u671b\u4f7f\u7528 Python \u63a5\u53e3\uff0c\u5247\u9700\u8981\u5b89\u88dd Python \u76f8\u95dc\u7684\u4f9d\u8cf4\u9805\uff1a<\/p>\n<pre><code>sudo apt-get install python-dev python-numpy python-pip\npip install protobuf<\/code><\/pre>\n<p>\u7136\u5f8c\uff0c\u7de8\u8b6f Python \u63a5\u53e3\uff1a<\/p>\n<pre><code>cd python\nfor req in $(cat requirements.txt); do pip install $req; done\ncd ..\nmake pycaffe<\/code><\/pre>\n<h3 id=\"7-%e6%b8%ac%e8%a9%a6%e5%ae%89%e8%a3%9d-OhhAhQyDiK\">7. \u6e2c\u8a66\u5b89\u88dd<\/h3>\n<p>\u5b89\u88dd\u5b8c\u6210\u5f8c\uff0c\u53ef\u4ee5\u901a\u904e\u904b\u884c\u4ee5\u4e0b\u547d\u4ee4\u4f86\u6e2c\u8a66 Caffe \u662f\u5426\u6b63\u5e38\u5de5\u4f5c\uff1a<\/p>\n<pre><code>cd examples\/mnist\npython train_lenet.py<\/code><\/pre>\n<p>\u5982\u679c\u4e00\u5207\u6b63\u5e38\uff0c\u60a8\u61c9\u8a72\u80fd\u770b\u5230\u8a13\u7df4\u904e\u7a0b\u7684\u8f38\u51fa\u3002<\/p>\n<h2 id=\"%e7%b8%bd%e7%b5%90-OhhAhQyDiK\">\u7e3d\u7d50<\/h2>\n<p>\u5728\u672c\u6587\u4e2d\uff0c\u6211\u5011\u4ecb\u7d39\u4e86\u5982\u4f55\u5728 Ubuntu 16.04 \u4e0a\u5b89\u88dd Caffe \u7684 GPU \u7248\u672c\u3002\u9019\u4e9b\u6b65\u9a5f\u6db5\u84cb\u4e86\u5f9e\u7cfb\u7d71\u66f4\u65b0\u5230\u7de8\u8b6f\u548c\u6e2c\u8a66\u7684\u6574\u500b\u904e\u7a0b\u3002\u900f\u904e\u4f7f\u7528 Caffe\uff0c\u60a8\u53ef\u4ee5\u5145\u5206\u5229\u7528 GPU \u7684\u8a08\u7b97\u80fd\u529b\uff0c\u9032\u884c\u9ad8\u6548\u7684\u6df1\u5ea6\u5b78\u7fd2\u6a21\u578b\u8a13\u7df4\u3002\u5982\u679c\u60a8\u9700\u8981\u7a69\u5b9a\u7684\u74b0\u5883\u4f86\u904b\u884c Caffe\uff0c\u8003\u616e\u4f7f\u7528 <a href=\"https:\/\/server.hk\">\u9999\u6e2fVPS<\/a> \u670d\u52d9\uff0c\u4ee5\u7372\u5f97\u66f4\u597d\u7684\u6027\u80fd\u548c\u652f\u6301\u3002<\/p>\n","protected":false},"excerpt":{"rendered":"<p>Learn how to install the GPU version of Caffe on Ubuntu 16.04 with step-by-step instructions for optimal performance in deep learning tasks.<\/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-51413","post","type-post","status-publish","format-standard","hentry","category-setup-tutorials"],"_links":{"self":[{"href":"https:\/\/server.hk\/cnblog\/wp-json\/wp\/v2\/posts\/51413","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=51413"}],"version-history":[{"count":1,"href":"https:\/\/server.hk\/cnblog\/wp-json\/wp\/v2\/posts\/51413\/revisions"}],"predecessor-version":[{"id":51414,"href":"https:\/\/server.hk\/cnblog\/wp-json\/wp\/v2\/posts\/51413\/revisions\/51414"}],"wp:attachment":[{"href":"https:\/\/server.hk\/cnblog\/wp-json\/wp\/v2\/media?parent=51413"}],"wp:term":[{"taxonomy":"category","embeddable":true,"href":"https:\/\/server.hk\/cnblog\/wp-json\/wp\/v2\/categories?post=51413"},{"taxonomy":"post_tag","embeddable":true,"href":"https:\/\/server.hk\/cnblog\/wp-json\/wp\/v2\/tags?post=51413"}],"curies":[{"name":"wp","href":"https:\/\/api.w.org\/{rel}","templated":true}]}}