Ubuntu16.04深度學習基本環境搭建，tensorflow , keras , pytorch , cuda

Ubuntu16.04安裝

參考https://blog.csdn.net/flyyufenfei/article/details/79187656

安裝nvidia驅動

sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt update

選擇安裝驅動
ubuntu-drivers devices

查看自己顯卡對應的驅動型號，可在nvidia官網查詢
sudo apt install nvidia-410

通過以下命令查看是否安裝成功

nvidia-smi

 

安裝cuda,可選方式3種

1.直接安裝cuda,較為復雜，鏈接 https://blog.csdn.net/qq_32113189/article/details/83275817

2.通過anaconda中安裝keras-gpu版本時候自動安裝（本文下邊是這種方式）

3.docker方式，配置一次較麻煩，但以后使用極為方便，非常推薦（不願意折騰的可忽略），鏈接 https://blog.csdn.net/ambm29/article/details/96151358

安裝anaconda

https://blog.csdn.net/weixin_43715458/article/details/100096496

 

 

至此，深度學習環境的基本條件具備，下面安裝深度學習框架

安裝keras

conda install keras-gpu

過程會自動安裝cuda、tensorflow、keras-gpu，缺點是不可選擇版本，可先安裝下來，然后再卸載不想用的版本，自己重裝對應庫的版本。

安裝tensorflow

conda install tensorflow-gpu
不可修改版本
pip install tensorflow-gpu==1.15.0
可修改版本

也可下載源文件安裝，查詢官網下載地址即可

安裝pytorch

去pytorch官網https://pytorch.org/查看conda安裝pytorch的命令

1.命令行安裝

去pytorch官網https://pytorch.org/查看conda安裝pytorch的命令

2.源碼安裝（下載源文件）

去pytorch官網https://pytorch.org/下載對應版本