ubuntu17.04 + cuda 8.0 + cudnn 六 + tensorflow 1.2
聽 聽 聽 聽 鍏徃涔颁簡涓€涓湇鍔″櫒鐢ㄤ簬鏈哄櫒瀛︿範 锛屼簬鏄紑濮嬩簡婕极瀹夎璺€?/p>
聽 聽 聽 聽 鏌ヨ祫鏂欏彂鐜?tensorflow gpu妯″紡鍦╱buntu绯荤粺涓婃瘮杈冨ソ璺戯紝灏辫绯荤粺閮ㄥ悓浜嬭浜唘buntu绯荤粺64浣嶃€傛満鍣ㄥ埌鎵嬪悗锛屽彂鐜拌鐨勬槸ubuntu14.04锛屾劅瑙夌増鏈暐浣庯紝鍗囩骇鍒?6.04銆傜劧鍚庡畨瑁卬vidia,鐢ㄥ懡浠?/p>
聽lspci | grep -i nvidia涔熸煡鍒颁簡锛屽畨瑁卌uda锛屼絾鏄竴鐩存姤閿?br>
聽寮€濮嬩互涓烘槸绯荤粺鐨勯棶棰橈紝鍙堟妸绯荤粺鍗囩骇鍒颁簡17.04锛岀劧骞跺嵉锛岃繕鏄姤涓€鏍风殑閿欍€傚悗鏉ラ棶浜嗘埓灏旂殑宸ヤ綔浜哄憳锛屼汉瀹惰鍙兘鏄嚎娌℃彃濂斤紝瀵艰嚧渚涚數涓嶈冻銆傚湪杩滅▼鎸囧涓嬶紝鍚屼簨杩炰笂浜嗗叧閿殑涓€鏍圭嚎锛屾灉鐒朵笉鎶ラ敊浜嗐€傚ソ浜嗭紝纭欢瀹屽锛屽紑鍚厤鐜涔嬭矾銆?/p>
聽
涓€鍒囦粠澶磋璧枫€?/p>
1.瀹夎nvidia椹卞姩
鍘诲畼缃?a target="_blank" href="http://www.nvidia.cn/Download/index.aspx?lang=cn">http://www.nvidia.cn/Download/index.aspx?lang=cn鎵鹃€傚悎鑷繁鐨勯┍鍔ㄣ€?/p>
聽鎸夌収瀹樼綉鐨勫畨瑁呮柟娉曞畨瑁呭嵆鍙€?/p>
(
sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt-get update聽
sudo apt-get install nvidia-367聽
sudo apt-get install mesa-common-dev聽
sudo apt-get install freeglut3-dev
鎴戝紑濮嬫寜鐓т互涓婂懡浠よ鐨?75锛屽悗鏉ョ増鏈笉瀵癸紝瀹樼綉鐨勮繖涓竴鍒囨甯?
2.瀹夎cuda
鍗歌浇cuda聽http://blog.****.net/u012436149/article/details/53163346
寮€濮嬫湰鐫€瑕佺敤鏈€鏂扮増鐨勫師鍒欙紝瀹夎浜哻uda9.0锛屽悗鏉ヨtensorlfow鍙戠幇cuda鐗堟湰澶珮浜嗐€傝8.0灏卞彲浠ャ€備笅杞借瀹樼綉https://developer.nvidia.com/cuda-downloads銆傛垜寮€濮?.0(9.0鐨勫湴鍧€鏄痳ealse-download鍦板潃https://developer.nvidia.com/cuda-release-candidate-download)鐨勬椂鍊欑敤鐨勬槸runfile瀹夎鏂规锛屽畨瑁呬竴鍒囨甯革紝浣嗘槸鍚庢潵鍙戠幇鐗堟湰涓嶅厑璁革紝涔嬪悗瑁?.0鐨勬椂鍊檙unfile鎬绘槸鎶ラ敊锛?strong>灏辨敼涓?/strong>deb鏂规浜嗭紝鎸夌収瀹樼綉鏂规硶瀹夎锛屼竴鍒囨甯?/strong>銆?/p>
濡傛灉閲囩敤runfile鏂规锛屾湁涓粏鑺傞渶瑕佹敞鎰?,鍦ㄨ闂營nstall NVIDIA Accelerated Graphics Driver fo Linux-xx鏃讹紝閫塶銆傚洜涓哄墠闈㈠凡缁忚浜嗗悎閫傜増鏈殑nvidia锛岃繖閲屽啀瑁呬細鏈変袱涓紙鑰屼笖杩欎釜鐗堟湰杈冧綆锛夛紝涔嬪悗鍦ㄨ窇cuda鐨剆amples鏃朵細鏈変竴涓繑鍥?8鐨勬姤閿欙紝鑻ユ槸鍥犱负鐗堟湰闂浼氳繑鍥炰竴涓?5鐨勬姤閿欍€?br> 聽瑁呭畬锛岃繘鍏?usr/local/cuda-8.0/samples/1_Utilities/deviceQuery 聽浣跨敤 make鍛戒护 锛岀劧鍚庤繍琛?/deviceQuery 聽 鍘诲畼缃戜笅杞?a target="_blank" href="https://developer.nvidia.com/cudnn">https://developer.nvidia.com/cudnn聽 鎴戠洰鍓嶇敤鐨勬槸6.0锛?017.9.4锛夊彲鑳借繃鍑犱釜鏈坱ensorflow灏辨敮鎸?.0涔熻涓嶅畾锛屾瘯绔?鏈堢殑鏃跺€欒繕鍦ㄧ敤5.1. tar -zxf cudnn-8.0-linux-x64-v6.0.tgz sudo cp include/cudnn.h /usr/local/cuda-8.0/include sudo cp lib64/libcudnn* /usr/local/cuda-8.0/lib64 4.tensorflow-gpu (1)瀹夎渚濊禆锛堟垜鐢ㄧ殑python3锛?/p>
sudo apt-get install python-pip python-dev # for Python 2.7 sudo apt-get install python3-pip python3-dev # for Python 3.n (2) pip install tensorflow-gpu # Python 2.7; GPU support pip3 install tensorflow-gpu # Python 3.n; GPU support 杩欎釜鎴戣姳浜嗗緢涔咃紝鍥犱负浠栨€绘槸瓒呮椂锛岃瘯浜嗗緢澶氶亶缁堜簬涓嬪ソ浜嗐€?/p>
聽 娴嬭瘯 $ python3 >>> import tensorflow >>> import tensorflow as tf >>> hello = tf.constant('haha,tensorflow') >>> sess = tf.session() >>> print(sess.run(hello)) 鍙傝€冧簡浠ヤ笅缃戝潃锛?/p>
http://www.linuxidc.com/Linux/2017-01/139319.htm http://blog.****.net/dream_an/article/details/74992346
聽
聽3.瑁卌udnn聽
聽鎴戝紑濮嬭鐨勬槸5.1锛岀劧鍚庤tensorflow鐨勬椂鍊欐姤閿欎簡锛孖mportError: libcudnn.so.6: cannot open shared object file: No such file or directory锛岄渶瑕?锛屾垜灏遍噸鏂拌浜?.鎵ц浠ヤ笅璇彞鍗冲彲銆?/p>
聽OK锛侊紒锛侊紒