Ubuntu上PyTorch的安装步骤是什么

以下是在Ubuntu上安装PyTorch的步骤：

使用pip安装

1. 更新系统包列表：sudo apt update。
2. 安装Python和pip：sudo apt install python3 python3-pip。
3. 创建并激活虚拟环境（可选）：python3 -m venv pytorch_env，source pytorch_env/bin/activate。
4. 安装PyTorch：
   • CPU版本：pip3 install torch torchvision torchaudio。
   • GPU版本：根据CUDA版本选择，如CUDA 11.7版本为pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu117。

使用conda安装

1. 安装Miniconda或Anaconda：从官网下载安装包进行安装。
2. 创建虚拟环境：conda create -n pytorch_env python=3.9。
3. 激活虚拟环境：conda activate pytorch_env。
4. 安装PyTorch：
   • CPU版本：conda install pytorch torchvision torchaudio cpuonly -c pytorch。
   • GPU版本：如CUDA 11.7版本为conda install pytorch torchvision torchaudio cudatoolkit=11.7 -c pytorch。

验证安装

在Python中输入import torch，print(torch.__version__)，print(torch.cuda.is_available())，查看版本号和CUDA是否可用。