Pytorch的安裝過程之pip、conda、Docker容器安裝

更新時間：2023年04月06日 10:59:23 作者：CV_Today

PyTorch是一個基于Python的開源深度學習框架，可用于訓(xùn)練和預(yù)測深度學習模型，PyTorch支持多種安裝方法，這篇文章主要介紹了Pytorch的安裝----pip、conda、Docker容器,需要的朋友可以參考下

前言

PyTorch是一個開源的深度學習框架，基礎(chǔ)教學從開始安裝學起，一步一個腳印。

提示：以下是本篇文章正文內(nèi)容，下面案例可供參考

一、Pyorch介紹

PyTorch是一個開源的深度學習框架，用于計算機視覺和自然語言處理等應(yīng)用程序的開發(fā)。它提供了一個靈活的編程模型，可以方便地進行模型定義、訓(xùn)練和分析。PyTorch采用了面向?qū)ο蟮木幊田L格，允許用戶定義自己的神經(jīng)網(wǎng)絡(luò)層和損失函數(shù)。它還提供了豐富的工具和庫，可以幫助用戶實現(xiàn)復(fù)雜的深度學習模型。 PyTorch還可以與其他框架進行無縫集成，比如與TensorFlow進行集成，從而可以讓用戶獲得最佳的深度學習體驗。

二、Pyorch安裝

由于pytorch的版本取決于使用的操作系統(tǒng)、Python版本和是否使用GPU因此下面先指定如下環(huán)境：
Ubuntu20.04
python版本可以在conda構(gòu)建的虛擬環(huán)境中自由設(shè)定，以python3.8.15為例
GPU選擇本機對應(yīng)的型號，以RTX2080TI為例支持CUDA11.6

1.pip安裝

首先，確保你已經(jīng)安裝了Python 3.8和pip。如果你還沒有安裝，可以在命令行中輸入以下命令來安裝：

 sudo apt install python3 python3-pip

接下來，安裝PyTorch的核心庫和CUDA支持：

 pip install torch torchvision
 pip install torch torchvision cudatoolkit=11.6 -f https://download.pytorch.org/whl/torch_stable.html

然后，安裝cuDNN，這是Nvidia提供的一個用于加速深度學習模型訓(xùn)練的庫：

 # 下載安裝包
 wget https://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu2004/x86_64/nvidia-machine-learning-repo-ubuntu2004_1.0.0-1_amd64.deb
 # 安裝安裝包
 sudo apt install ./nvidia-machine-learning-repo-ubuntu2004_1.0.0-1_amd64.deb
 # 更新源
 sudo apt update
 # 安裝cuDNN
 sudo apt install --no-install-recommends libcudnn8=8.2.0.30-1+cuda11.6 libcudnn8-dev=8.2.0.30-1+cuda11.6

最后，運行以下代碼來驗證安裝是否成功：

 import torch
 print(torch.__version__)

如果安裝成功，那么會打印出當前安裝的PyTorch版本號。

注意：在安裝過程中，如果遇到問題，可以參考PyTorch官網(wǎng)的安裝指南：https://pytorch.org/get-started/locally/。

2.conda安裝

首先，確保你已經(jīng)安裝了Python 3.8和Conda。如果你還沒有安裝，可以在命令行中輸入以下命令來安裝：

 # 下載安裝包
 wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh
 # 安裝
 bash Miniconda3-latest-Linux-x86_64.sh

接下來，安裝PyTorch的核心庫和CUDA支持：

 conda install pytorch torchvision cudatoolkit=11.6 -c pytorch

然后，安裝cuDNN，這是Nvidia提供的一個用于加速深度學習模型訓(xùn)練的庫：

 # 下載安裝包
 wget https://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu2004/x86_64/nvidia-machine-learning-repo-ubuntu2004_1.0.0-1_amd64.deb
 # 安裝安裝包
 sudo apt install ./nvidia-machine-learning-repo-ubuntu2004_1.0.0-1_amd64.deb
 # 更新源
 sudo apt update
 # 安裝cuDNN
 sudo apt install --no-install-recommends libcudnn8=8.2.0.30-1+cuda11.6 libcudnn8-dev=8.2.0.30-1+cuda11.6

最后，運行以下代碼來驗證安裝是否成功：

 import torch
 print(torch.__version__)

如果安裝成功，那么會打印出當前安裝的PyTorch版本號。

注意：在安裝過程中，如果遇到問題，可以參考PyTorch官網(wǎng)的安裝指南：https://pytorch.org/get-started/locally。

3.Docker容器安裝

首先，確保已經(jīng)安裝了Docker。如果你還沒有安裝，可以在命令行中輸入以下命令來安裝：

 sudo apt install docker.io

接下來，下載PyTorch的Docker鏡像：

 sudo docker pull pytorch/pytorch:1.8.1-cuda11.6-cudnn8-devel

然后，運行Docker容器：

sudo docker run -it pytorch/pytorch:1.8.1-cuda11.6-cudnn8-devel

進入Docker容器后，就可以在容器中運行PyTorch的Python代碼了。

注意：使用Docker容器安裝PyTorch的優(yōu)點是簡單、快捷，但是也有一些缺點，比如容器隔離的環(huán)境與宿主機的環(huán)境存在一定差異，如果需要使用宿主機上的文件，需要使用Docker的“掛載”功能，這需要一定的Docker知識和技巧。建議在安裝PyTorch前，先了解Docker的基本操作方法。