Ubuntu 20.04 下的 PyTorch（CPU版）安装及配置

最新推荐文章于 2024-06-15 20:08:29 发布

Cybersonix

最新推荐文章于 2024-06-15 20:08:29 发布

阅读量5k

点赞数 5

文章标签： pytorch python ubuntu

本文链接：https://blog.csdn.net/haiyunzhiqiu/article/details/119342760

版权

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

一、通过 Anaconda 安装 Python

进入 Anaconda3 官网，下载 Anaconda3 Individual Edition for Linux 。
进入文件下载目录，在终端执行如下命令，开始安装Anaconda。
```
bash Anaconda3-2021.05-Linux-x86_64.sh
```
接下来根据安装提示，输入"yes"或者按回车即可。期间会提示选择安装路径，如果没有特殊要求，建议使用默认路径（~/anaconda3），然后就开始安装。
按回车键，进入注册信息页面。
按q跳过阅读，yes。
默认安装在用户目录下，直接回车即可安装；若想自定义安装目录，直接输入安装目录，回车即可。
出现提示信息：Do you wish the installer to initialize Anaconda3 by running conda init ? 输入 no，回车。
看到下面的提示信息说明安装完成。
将Anaconda添加到用户环境变量中。开启一个新的终端，输入以下指令：

安装vim编辑器（仅在没有安装 vim 时使用）
```
sudo apt install vim
```
使用vim编辑器打开 .bashrc 环境变量文件
```
vim ~/.bashrc
```
跳转至文件末尾，输入 “o” 在当前行的下方新打开一个行，并切换到插入模式，插入下面的命令，然后输入 “Esc” “:wq” “Enter” 保存文本文件。
```
export PATH="/home/sun/anaconda3/bin:$PATH"
```
输入下面的命令使编辑之后使环境变量生效。
```
source ~/.bashrc
```
检查是否安装成功。

查看版本。在终端中输入：
```
conda --version
```
或
```
conda -V
```
出现下面的提示信息，说明安装成功。

输入
```
python3
```
出现下面的提示信息，说明是 Anaconda 安装的版本。

二、安装 PyTorch

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

如上图所示，复制" Run this Command "中的命令到终端，运行。
```
conda install pytorch torchvision torchaudio cpuonly -c pytorch
```
提示安装及升级信息，输入 “y”。下载过程较慢，请耐心等待。

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