Ubuntu 20.04 下的 PyTorch(CPU版)安装及配置

运行环境:Ubuntu 20.04 LTS(虚拟机)
硬件配置:Intel i7-9750H CPU + NVIDIA GeForce GTX 1650 (Notebook) GPU

一、通过 Anaconda 安装 Python

  1. 进入 Anaconda3 官网,下载 Anaconda3 Individual Edition for Linux 。

  2. 进入文件下载目录,在终端执行如下命令,开始安装Anaconda。

    bash Anaconda3-2021.05-Linux-x86_64.sh
    

    接下来根据安装提示,输入"yes"或者按回车即可。期间会提示选择安装路径,如果没有特殊要求,建议使用默认路径(~/anaconda3),然后就开始安装。

  3. 按回车键,进入注册信息页面。

  4. 按q跳过阅读,yes。

  5. 默认安装在用户目录下,直接回车即可安装;若想自定义安装目录,直接输入安装目录,回车即可。

  6. 出现提示信息:Do you wish the installer to initialize Anaconda3 by running conda init ? 输入 no,回车。

  7. 看到下面的提示信息说明安装完成。

  8. 将Anaconda添加到用户环境变量中。开启一个新的终端,输入以下指令:

    安装vim编辑器(仅在没有安装 vim 时使用)

    sudo apt install vim
    

    使用vim编辑器打开 .bashrc 环境变量文件

    vim ~/.bashrc
    

    跳转至文件末尾,输入 "o" 在当前行的下方新打开一个行,并切换到插入模式,插入下面的命令,然后输入 "Esc" ":wq" "Enter" 保存文本文件。

    export PATH="/home/sun/anaconda3/bin:$PATH"
    

    输入下面的命令使编辑之后使环境变量生效。

    source ~/.bashrc
    
  9. 检查是否安装成功。

    查看版本。在终端中输入:

    conda --version
    

    conda -V
    

    出现下面的提示信息,说明安装成功。

    输入

    python3
    

    出现下面的提示信息,说明是 Anaconda 安装的版本。

二、安装 PyTorch

  1. 进入 Pytorch 官网。下滑找到配置列表,按照图示参数进行选择。

    如上图所示,复制" Run this Command "中的命令到终端,运行。

    conda install pytorch torchvision torchaudio cpuonly -c pytorch
    

    提示安装及升级信息,输入 "y"。下载过程较慢,请耐心等待。

    出现下面的提示信息,说明安装完成。


2021年7月28日更新