Ubuntu16.04服務器安裝GPU版本的tensorflow1.5.0、pytorch0.4、pytorch1.0

1、非root用戶下安裝cuda9.0和cudnn7.0.5

與tensorflow1.5.0適配的是cuda9.0和cudnn7.0.5,服務器非root用戶參考https://blog.csdn.net/weixin_41278720/article/details/81255265安裝相應版本。

2、安裝Anaconda

主要參考https://blog.csdn.net/hgdwdtt/article/details/78633232

  1. 首先從清華鏡像網站上下載合適的源文件,比如下載Anaconda3-5.0.1-Linux-x86_64.sh
  2. 將下載的源文件通過FileZilla上傳到服務器個人目錄下,比如爲/home/zhang
  3. 通過xshell連接到服務器終端,運行bash /home/zhang/Anaconda3-5.0.1-Linux-x86_64.sh,然後一直yes、一直enter,安裝過程中會提示設置安裝位置

驗證anaconda是否安裝成功:

python
>>>import numpy #查看是否能夠運行,numpy爲anaconda內置python庫。

3、安裝tensorflow

3.1 首先創建tesorflow環境

可以起名爲tf1.5或者tensorflow1.5等等,怎麼方便怎麼來。這一步一定要做!!!千萬不能忘了!

conda create -n tensorflow1.5 python=3.6

此處python=3.6表示python的版本。

3.2 激活tensorflow環境

conda activate tensorflow1.5

經過這步後,將會進入tensorflow1.5環境下,終端顯示的目錄前面將會有個(tensorflow1.5),後面安裝tensorflow或者其他package都需要進入環境。

3.3 安裝GPU版本的tensorflow

pip install tensorflow-gpu==1.5.0

3.4 驗證是否安裝成功

python  #啓動python環境
# 驗證TensorFlow是否安裝成功
>>> import tensorflow as tf 
>>> hello = tf.constant('Hello, TensorFlow!')
>>> sess = tf.Session()
>>> print sess.run(hello)
Hello, TensorFlow! # 恭喜!安裝成功!

>>> sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))
# 輸出裏有 GPU的名稱,顯存等信息,表示tensorflow可以使用GPU了

注:可以通過conda remove -n tensorflow1.5 --all刪除不想要的環境。

3.5 安裝後續其他需要的package

在tensorflow1.5環境中,可以通過conda install xxx或者pip install pillow==5.2.0安裝opencv-python、scikit-image等依賴包。

4、安裝pytorch

與安裝tensorflow的步驟類似,參考https://blog.csdn.net/amateur_hy/article/details/90716411

  • 創建虛擬環境
conda create -n pytorch0.4 python=3.5
  • 進入虛擬環境
conda activate pytorch0.4
  • 安裝pytorch

去pytorch官網https://pytorch.org/,根據系統配置選擇合適的版本,按照官網給出的命令,完成安裝。

對於pytorch0.4版本,運行:

conda install pytorch=0.4.1 cuda90 -c pytorch
  • 安裝torchvision
conda install torchvision

在pytorch0.4環境下,同樣可以用conda install xxx或者pip install xxx安裝其他依賴項。

對於pytorch1.0或者其他版本,只需要在pytorch官網上選擇對應版本的指令,其他操作都類似。

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章