Linux下PyTorch的安装

一、Linux安装PyTorchGPU

首先，我们需要先安装NVIDIA CUDA和cuDNN，然后再安装PyTorchGPU。以下是安装步骤：

1、安装NVIDIA CUDA，按照自己显卡的型号选择相应的CUDA版本，并按照官方文档进行安装。安装完成后，可以通过以下命令查看CUDA版本号：

$ nvcc --version

2、安装cuDNN，也需按照CUDA版本选择相应的cuDNN版本，并按照官方文档进行安装。

3、安装PyTorchGPU。可以使用conda或pip进行安装，以下是使用conda进行安装的方法：

$ conda install pytorch torchvision torchaudio cudatoolkit=11.0 -c pytorch

这里的cudatoolkit版本要与安装的CUDA版本相对应。

二、Linux安装PyTorch一直Solving

在使用conda安装PyTorch过程中，有时会发生一直Solving的情况。这时可以检查一下网络连接情况，或尝试更改conda的镜像源，以下是设置镜像源的方法：

1、备份原本的.condarc文件（如果存在）：

$ mv ~/.condarc ~/.condarc.bak

2、新建.condarc文件并添加清华镜像源：

$ echo "channels:" > ~/.condarc
$ echo "  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/" >> ~/.condarc
$ echo "  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/menpo/" >> ~/.condarc
$ echo "  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/" >> ~/.condarc
$ echo "  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/" >> ~/.condarc
$ echo "  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/" >> ~/.condarc

3、清除conda缓存，使设置生效：

$ conda clean --all

4、重新安装PyTorch：

$ conda install pytorch torchvision torchaudio cudatoolkit=11.0 -c pytorch

三、Linux安装PyTorch的步骤

除了使用conda进行安装外，还可以使用pip进行安装，以下是安装步骤：

1、安装NVIDIA CUDA和cuDNN，方法同上述PyTorchGPU的安装步骤。

2、安装PyTorch。可以使用以下命令进行pip安装：

$ pip install torch torchvision torchaudio

要注意的是，这里安装的是CPU版本的PyTorch，如果需要安装GPU版本的，需要指定cudatoolkit版本，例如：

$ pip install torch torchvision torchaudio -f https://download.pytorch.org/whl/cu110/torch_stable.html

四、Linux安装

如果你还没有安装Linux系统，可以根据自己的需求选择Ubuntu、CentOS、Debian等常见的Linux版本。以下介绍Ubuntu系统的安装：

1、首先从官网下载Ubuntu系统镜像文件。

2、将镜像文件写入USB闪存盘，准备通过它来安装Ubuntu。可以使用dd命令或者工具如UNetbootin进行写入。

3、确保计算机设置为从USB启动，在计算机开机时按F2或其他键进入Boot Menu，选择从USB启动。

4、在启动盘中选择安装Ubuntu，并完成设置。

五、PyTorch安装教程

除了在Linux系统中安装PyTorch外，还可以在Windows、macOS等操作系统中安装，具体安装方法可以参考PyTorch官方文档中的教程。

六、conda安装PyTorch

conda是一个常用的包管理工具，在安装PyTorch时也可以使用conda进行安装。

1、首先需要安装Anaconda，可以从官网下载相应版本的Anaconda安装包进行安装。

2、在Anaconda Prompt中输入以下命令来创建一个新的Python环境，并在该环境中安装PyTorch：

$ conda create -n myenv python=3.7
$ conda activate myenv
$ conda install pytorch torchvision torchaudio cpuonly -c pytorch

这里的cpuonly表示安装的是CPU版本的PyTorch。如果需要安装GPU版本的，可以省略这个参数，并按照自己的CUDA版本指定cudatoolkit参数。

七、Python安装torch

如果不想使用conda进行安装，还可以直接用pip安装torch。

1、首先需要安装Python，可以从官网下载相应版本的Python安装包进行安装。

2、在命令行中使用以下命令进行安装：

$ pip install torch torchvision torchaudio

同样地，要注意安装CPU版本的PyTorch和GPU版本的PyTorch的命令略有不同，详情请参见官方文档。

八、Linux安装DNS选取

为了加快软件下载速度，可以选择一些国内的DNS服务器，以下是三个常见的DNS服务器地址：

1、阿里DNS：

223.5.5.5
223.6.6.6

2、百度DNS：

180.76.76.76
2400:cb00:2048:1::1

3、CNNIC DNS：

1.2.4.8
210.2.4.8
101.224.22.8
218.30.118.6
218.30.118.7
180.76.76.219
2400:da00::6666

在Linux系统中，可以通过编辑/etc/resolv.conf文件来设置DNS服务器地址：

$ sudo vim /etc/resolv.conf

在该文件中添加以下内容：

nameserver DNS服务器地址

多个DNS服务器地址可以按照上述格式依次添加。保存退出后，可以使用以下命令查看当前系统所用的DNS服务器：

$ cat /etc/resolv.conf