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