PyTorch在Ubuntu上的安装教程有哪些

以下是一些PyTorch在Ubuntu上的安装教程：

使用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 11.7为例）：pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu117。

使用conda安装

1. 安装Miniconda或Anaconda：从官网下载安装脚本并运行，如bash Miniconda3-latest-Linux-x86_64.sh，然后source ~/.bashrc。
2. 创建虚拟环境：conda create -n pytorch_env python=3.8，再conda activate pytorch_env。
3. 安装PyTorch：
   • CPU版本：conda install pytorch torchvision torchaudio cpuonly -c pytorch。
   • GPU版本（以CUDA 11.7为例）：conda install pytorch torchvision torchaudio cudatoolkit=11.7 -c pytorch。

安装完成后，可通过import torch，print(torch.__version__)和print(torch.cuda.is_available())验证安装是否成功。