ubuntu16.04 + cuda8.0安裝
本文記得不夠細致,有待細查
下載cuda.run,我自己的網盤也有
網上下載cuda.run
1.安裝cuda
chmod +x ./sudo sh cuda_8.0.61_375.26_linux.run
sudo ./sudo sh cuda_8.0.61_375.26_linux.run
(1)運行后,先出來協議0%,按空格下翻,最后選擇accept,
(2)如果 nvidia的驅動已經安裝了,選擇no。
(3)后面就yes 或者回車,直到安裝成功
2.安裝cudnn
2.安裝cuDNN
下載cuDNN,從https://developer.nvidia.com/rdp/cudnn-download,注冊,然后下載,
注意事項:
有可能由于最新cuDNN不穩定,導致后續caffe工程編譯失敗,報出如下錯誤,詳情參考第三個鏈接,這時可以回退一下,下載較新版本的cuDNN;
拷貝cuDNN庫文件到cuda目錄下:
tar -zxvf cudnn-7.0-linux-x64-v4.0-prod.tgz
cd cuda
sudo cp lib64/* /usr/local/cuda/lib64/
sudo cp include/cudnn.h /usr/local/cuda/include/
6、測試cuda的samples
cd /usr/local/cuda/samples/1_Utilities/deviceQuery make sudo ./deviceQuery
5、執行
cd /usr/local/cuda/lib64/ sudo rm -rf libcudnn.so libcudnn.so.6 #刪除原有動態文件 sudo ln -s libcudnn.so.6.0.21 libcudnn.so.6 #生成軟鏈接 sudo ln -s libcudnn.so.6 libcudnn.so #生成軟鏈接
sudo ldconfig
注意??!軟鏈接中的數字6.0.21等參照文件夾lib64中原有的軟鏈接名
設置環境變量,在/etc/profile中添加CUDA環境變量,
sudo gedit /etc/profile
在打開的文件中加入如下兩句話
export PATH=/usr/local/cuda/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH
3、python
輸入python后輸入以下示例
>>>import tensorflow as tf
>>>hello = tf.constant('Hello, TensorFlow!')
>>>sess = tf.Session()
>>>print sess.run(hello)
Hello, TensorFlow!
>>>a = tf.constant(10)
>>>b = tf.constant(32)
>>>print sess.run(a+b)
42
>>>
5、測試成功接下來首先退出python ?按快捷鍵Ctrl+D
Ubuntu
版權聲明:本文內容由網絡用戶投稿,版權歸原作者所有,本站不擁有其著作權,亦不承擔相應法律責任。如果您發現本站中有涉嫌抄襲或描述失實的內容,請聯系我們jiasou666@gmail.com 處理,核實后本網站將在24小時內刪除侵權內容。