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

2023-07-29,,

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

Ubuntu16.04深度学习基本环境搭建,tensorflow , keras , pytorch , cuda的相关教程结束。

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

下载本文的Word格式文档,以方便收藏与打印。