在Ubuntu上如何开始学习PyTorch

在Ubuntu上开始学习PyTorch，你可以按照以下步骤进行：

安装PyTorch

使用pip安装

1. 更新包列表：

sudo apt update

2. 安装必要的依赖项：

sudo apt install -y build-essential cmake git libopenblas-dev liblapack-dev libjpeg-dev libpng-dev libtiff-dev libavcodec-dev libavformat-dev libswscale-dev python3-dev python3-pip

3. 安装Python3和pip（如果尚未安装）：

sudo apt install -y python3 python3-pip

4. 创建一个新的虚拟环境（可选）：

python3 -m venv pytorch_env
source pytorch_env/bin/activate

5. 安装PyTorch： 根据你的CUDA版本选择合适的命令。例如，使用CPU版本：

pip3 install torch torchvision torchaudio

或者使用CUDA 11.3版本：

pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu113

使用conda安装（推荐）

1. 安装Miniconda或Anaconda：

wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh -O miniconda.sh
bash miniconda.sh -b

2. 创建一个新的conda环境：

conda create -n pytorch_env python=3.8

3. 激活环境：

conda activate pytorch_env

4. 安装PyTorch： 使用CPU版本：

conda install pytorch torchvision torchaudio cpuonly -c pytorch

或者使用CUDA 11.3版本：

conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch

验证安装

在Python中输入以下代码来验证PyTorch是否安装成功：

import torch
print(torch.__version__)
print(torch.cuda.is_available())  # 如果安装了GPU版本，应该返回True

学习资源

• PyTorch官方文档：PyTorch官方文档
• 快速入门：提供简洁的教程帮助新手快速上手PyTorch。
• 教程中心：从基础张量操作到高级应用（如分布式训练），含代码示例和Colab在线运行环境。

通过以上步骤，你应该能够在Ubuntu系统上成功安装PyTorch，并开始你的深度学习学习之旅。如果有任何问题，可以查阅PyTorch官方文档或社区论坛以获取帮助。