ubuntu20.04安装pytorch

在 Ubuntu 20.04 中安装 PyTorch 的步骤如下:

1. 安装 Anaconda（如果您还没有安装），可以在终端中使用以下命令安装:

wget https://repo.anaconda.com/archive/Anaconda3-2021.07-Linux-x86_64.sh
bash Anaconda3-2021.07-Linux-x86_64.sh

2. 创建一个新的 Anaconda 环境，如：

conda create -n pytorchenv python=3.8

3. 激活该环境

conda activate pytorchenv

4. 安装 PyTorch

conda install pytorch torchvision -c pytorch

5. 安装 torchaudio

conda install torchaudio -c pytorch

6. 若需要 cudatoolkit 支持,在安装 pytorch时添加 cudatoolkit 版本, 例如安装 CUDA 11.0 版本:

conda install pytorch torchvision cudatoolkit=11.0 -c pytorch

7. 验证安装是否成功,可以在终端中使用以下命令

python -c "import torch;print(torch.__version__)"

若看到相应的版本号，说明安装成功。