PyTorch如何在Ubuntu中配置环境

以下是在Ubuntu中配置PyTorch环境的步骤：

使用pip安装

1. 更新系统：sudo apt update && sudo apt upgrade -y。
2. 安装Python和pip：sudo apt install python3 python3-pip -y。
3. 创建并激活虚拟环境（可选）：python3 -m venv pytorch_env，source pytorch_env/bin/activate。
4. 安装PyTorch：根据CUDA版本选择命令，如无GPU用pip3 install torch torchvision torchaudio；有GPU则用pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cuXXX（XXX为CUDA版本号）。
5. 验证安装：在Python解释器中输入import torch，print(torch.__version__)，print(torch.cuda.is_available())。

使用conda安装

1. 安装Miniconda或Anaconda：从官网下载安装包，如bash Miniconda3-latest-Linux-x86_64.sh。
2. 创建新环境：conda create -n pytorch_env python=3.8。
3. 激活环境：conda activate pytorch_env。
4. 安装PyTorch：根据CUDA版本选择，如conda install pytorch torchvision torchaudio cpuonly -c pytorch或conda install pytorch torchvision torchaudio cudatoolkit=XXX -c pytorch（XXX为CUDA版本号）。
5. 验证安装：同pip安装的验证方法。