Ubuntu16.04深度学习基本环境搭建，tensorflow , keras , pytorch , cuda

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/下载对应版本