Ubuntu深度学习环境搭建：Cuda 11 + PyTorch / TensorFlow

Ubuntu深度学习环境搭建包含如下的步骤：

1.安装ubuntu 18.04 / centos 7.5 系统

2.安装cuda / cudnn

3.安装anaconda3 / PyTorch / TensorFlow

1.安装ubuntu 18.04

1） 在安装之前，需要检测cuda版本，cudnn的支持情况，查看cudnn需要登录（注册）nvidia的网站，可以看到cudnn目前支持到18.04，很快就会有20.04.

 

2）下载Cuda 11

按照官方的方法wget下载，或者直接在浏览器里吧这个http:// 链接放在网络上就能下载，大概2.9GB，下载完需要很久的时间

wget http://developer.download.nvidia.com/compute/cuda/11.0.2/local_installers/cuda_11.0.2_450.51.05_linux.run

3）cudnn选择安装

大概331MB，需要登录nvidia网站

cuDNN Runtime Library for Ubuntu18.04 x86_64 (Deb)

4）制作启动盘，有些使用UltraISO就可以，有些需要Rufus制作

5）安装Ubuntu 18.04操作系统

首先选择一个U盘，最好比较新的U盘

下载Ubuntu 18.04 server/desktop

使用Rufus或者UltraISO刻录操作系统

进入电脑/服务器的BIOS，选择从U盘进入

分区、安装、可以参考网上的流程