ubuntu安装pytorch怎么使用

要在Ubuntu上安装PyTorch并使用它，您可以按照以下步骤进行操作：

1. 安装Anaconda：在终端中执行以下命令，下载并安装Anaconda，这是一个用于管理Python环境的工具。

   wget https://repo.anaconda.com/archive/Anaconda3-2021.05-Linux-x86_64.sh
   chmod +x Anaconda3-2021.05-Linux-x86_64.sh
   ./Anaconda3-2021.05-Linux-x86_64.sh
   

   在安装过程中，按照提示进行配置。安装完成后，执行source ~/.bashrc命令，使Anaconda生效。

2. 创建新的Anaconda环境：在终端中执行以下命令，创建一个名为pytorch的新环境，并激活该环境。

   conda create -n pytorch python=3.8
   conda activate pytorch
   

3. 安装PyTorch：在激活的pytorch环境中，执行以下命令，安装PyTorch。

   conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch -c conda-forge
   

4. 验证安装：在Python交互式环境中，执行以下命令，验证PyTorch是否安装成功。

   import torch
   print(torch.__version__)
   

   如果能够成功输出PyTorch的版本号，则表示安装成功。

现在，您可以使用PyTorch进行深度学习任务了。可以查阅PyTorch的官方文档来学习如何使用PyTorch的各种功能和API。