ubuntu18.04如何安裝tensorflow gpu

1.安裝cuda和cudnn

第一個是安裝驅動

根據需要安裝的cuda版本,安裝對應的驅動,比如cuda10.1所需要的最小顯卡驅動版本爲418,我之前安裝的415版本的驅動就不可以使用,所以需要卸載驅動(如果卸載驅動之後出現一些很奇怪的界面,無法用boot-repair修復的話,還是重裝系統吧,感覺這樣比較快)

sudo apt-get remove --purge nvidia*

之後禁用nouveau驅動一定要安裝下文所述,否則會出現無法登陸進入桌面的情況

如果出現無法登陸桌面的情況,嘗試重裝系統https://gummary.github.io/2018/11/04/install-cuda-ubuntu/ 

或者使用boot-repair進行修復(修復可能要進行兩次,但也比重裝系統快!)https://blog.csdn.net/qq413187589/article/details/43273011

接着安裝cudn(一定要注意版本對應的顯卡驅動版本)

https://blog.csdn.net/qq_33200967/article/details/80689543#CUDNN_205

https://blog.csdn.net/yjt1325/article/details/84641062

2.安裝anconda2(博客的鏈接是安裝anconda3,但由於我使用的代碼是基於python2,所以這裏我安裝的仍然是anaconda2)

https://blog.csdn.net/u012318074/article/details/77074665

注意安裝完anaconda之後應該在~/.bashrc中導入環境變量

export PATH=/home/tj816/anaconda3/bin:$PATH

 

3. 安裝tensorflow

第一個比較好,第二個作爲參考

https://blog.csdn.net/Angela_qin/article/details/80921688

https://blog.csdn.net/m0_37864814/article/details/82112029

4.注意每次使用tensorflow的時候,都需要source activate tf

測試tf是否安裝成功

https://blog.csdn.net/jiang_z_q/article/details/73264561

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