Ubuntu安装Pytorch-gpu的简易步骤

寒假来了，想做个图像识别的demo，先把基本环境配置起来。这是一篇纯文字的、流水账式的记录。

• Step1：把电脑的操作系统换成Ubuntu20.04。这一部比较简单，先去ubuntu的官网下载一个iso的镜像文件，然后用UltraISO制作一个U盘的ubuntu安装盘。
• Step2：重新启动电脑，进入BIOS设置为U盘启动，然后安装Ubuntu系统；可以选择把整个磁盘擦除再安装、也可以选择保留原来的系统安装双系统。因为，我的这个电脑没有作其它用，所以直接格式化硬盘安装的。
• Step3：安装完Ubuntu之后，首先安装官方的显卡驱动。可以直接去nvida官网下载对应的linux的驱动，然后安装。比较简单的办法，就是直接用Ubuntu自带的Additional Drivers选择官方驱动按安装，选择的时候选一个非test的最高版本。
• Step4：安装完显卡驱动后，禁用ubuntu自带的显卡驱动nouveau，然后重新启动。
• Step5：安装Anaconda，去Anaconda的官网安装，当然也可以安装一个miniconda。安装完成后，可以改一下conda的源。
• Step6：创建一个专门用来作图像识别的虚拟环境，使用命令“conda create -n FGIA"。然后使用命令“conda activate FGIA"激活这个虚拟环境。
• Step7：在FGIA下使用conda命令安装Pytorch，在Pytorch的官网（PyTorch）选择合适自己电脑环境和安装方式的pytorch，然后将网站上生成的conda安装命令复制到自己电脑上运行。例如：“conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch”。此处注意：尽量安装新的版本，如果版本安装不对可能回导致后面torch的cuda不能正常使用。
• Step8：验证pytorch是否安装成功。在安装了Pytorch的虚拟环境进入Python，然后输入命令“import torch"不会报错，而且下面这些代码返回正常。

>>> import torch
>>> print(torch.__version__)
1.10.1
>>> print(torch.cuda.is_available())
True
>>> 

• Step9：安装Pycharm或者直接用anaconda自带的spyder等进行pytorch的开发。