安裝顯示卡驅動
安裝cuda 10.1.2
cuda官網 :cuda-10.1
根據官網上的安裝過程安裝
安裝完成後設定環境變數, 在終端輸入wget
sudo mv cuda-ubuntu1804.pin /etc/apt/preferences.d/cuda-repository-pin-600
wget
sudo dpkg -i cuda-repo-ubuntu1804-10-1-local-10.1.243-418.87.00_1.0-1_amd64.deb
sudo apt-key add /var/cuda-repo-10-1-local-10.1.243-418.87.00/7fa2af80.pub
sudo apt-get update
sudo apt-get -y install cuda
sudo gedit ~/.bashrc
在檔案中最後一行新增
export path=/usr/local/cuda-10.1/bin\$}
測試安裝是否成功(需要較長時間):
編譯完成後繼續輸入cd /usr/local/cuda-10.1/samples
sudo make
如果顯示「result = pass」表示安裝成功, 本人是重啟完系統後才顯示「result = pass」的.cd /usr/local/cuda-10.1/samples/1_utilities/devicequery
sudo make
./devicequery
3. 安裝cudnn
解壓cudnn
tar zxvf cudnn-10.1-linux-x64-v7.6.3.30.tgz
解壓完成後會得到乙個cuda檔案
複製include和lib64資料夾的檔案到相應目錄,具體路徑根據自己情況設定
在終端進入cuda檔案, 輸入
安裝tensorflow-gpu版sudo cp ./include/cudnn.h /usr/local/cuda-10.1/targets/x86_64-linux/include
sudo cp ./lib64/libcudnn* /usr/local/cuda-10.1/targets/x86_64-linux/lib/
sudo chmod a+r /usr/local/cuda-10.1/targets/x86_64-linux/include/cudnn.h
sudo chmod a+r /usr/local/cuda-10.1/targets/x86_64-linux/lib/libcudnn*
pip install tensorflow-gpu
測試**
import tensorflow as tf
with tf.device(
'/cpu:0'):
a = tf.constant(
[1.0
,2.0
,3.0
], shape=[3
], name=
'a')
b = tf.constant(
[1.0
,2.0
,3.0
], shape=[3
], name=
'b')
with tf.device(
'/gpu:1'):
c = a + b
# 注意:allow_soft_placement=true表明:計算裝置可自行選擇,如果沒有這個引數,會報錯。
# 因為不是所有的操作都可以被放在gpu上,如果強行將無法放在gpu上的操作指定到gpu上,將會報錯。
sess = tf.session(config=tf.configproto(allow_soft_placement=
true
, log_device_placement=
true))
# sess = tf.session(config=tf.configproto(log_device_placement=true))
sess.run(tf.global_variables_initializer())
(sess.run(c)
)
[2. 4. 6.]
參考:
1.ubuntu16.04配置tensorflow的gpu深度學習環境
2.tensorflow-gpu測試**
在windows環境下搭建TensorFlow
安裝python python 3.6.1 amd64.exe 加入python.exe pip pip3加入環境變數 在cmd下執行 pip3 install upgrade tensorflow import tensorflow as tf hello tf.constant hello,te...
深度學習 Ubuntu上安裝tensorflow
ubuntu上安裝tensorflow非常簡單,但還是會踩到一點小坑。本文假設你的ubuntu已經安裝好,然後安裝tensorflow過程如下 1.輸入 su 回車,輸入密碼 2.安裝pip sudo apt get install python pip 3.安裝tensorflow sudo pi...
採用ubuntu系統來安裝tensorflow
最近在學習google新開源的深度學習框架tensorflow。發現安裝它的時候,需要依賴python2.7.x 我之前一直使用的linux是centos。而centos不更新了,裡面的自帶的python一般都是python2.6以下的。不僅如此,系統裡面很多元件又依賴python2.6,所以導致你...