服务器安装pytorch

提示：文章写完后，目录可以自动生成，如何生成可参考右边的帮助文档

文章目录

一、前言

二、安装步骤

1.安装anaconda

2.pytorch安装

三、总结

---

一、前言

本文记录深度学习小白在阿里云服务器上（linux系统）安装pytorch的全过程，仅作记录，供个人今后学习用。

二、安装步骤

1.安装anaconda

在anaconda的官网Anaconda | Anaconda Distribution中选择适合自己系统的anaconda版本，复制下载链接。在服务器终端中输入wget+空格+下载链接。

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

下载后在当前文件夹下看到Anaconda3-2022.02-linux-x86_64.sh文件

​

 输入（尽量不要在root下运行）

sh Anaconda3-2022.05-Linux-x86_64.sh

anaconda就安装成功了。

​

 安装成功后还需要配置环境变量,编辑~/.bashrc文件，这是本地的环境变量配置，我用的非root用户运行（vi ~/.bashrc）。在文件最后写入

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

具体示例如下图：

​

写入成功后，输入命令使文件生效，这样anaconda就配好啦~（可以输入conda进行确认）

yuyu@pyhome:~$ source ~/.bashrc

2.pytorch安装

首先，需要创建pytorch的一个虚拟环境,创建完成后进入该环境。

conda create -n (环境名) python=3.9

(base) yuyu@pyhome:~$ conda activate pytorch
(pytorch) yuyu@pyhome:~$ conda install pytorch torchvision torchaudio cpuonly -c pytorch

安装过程中可能会出现部分包无法安装的现象，我这里是修改了镜像源，输入命令vi ~/.condarc，（部分人可能找不到这个文件，可以输入conda config来初始化下）修改里面的内容为：

channels:
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/menpo/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/bioconda/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
  - http://mirror.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
  - http://mirror.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
  - defaults
show_channel_urls: true

修改过后输入conda clean -i 对之前的缓存进行清理。

安装完成后，进入python命令行界面输入import torch ，安装成功。

---

三、总结

本文仅仅简单介绍了ubuntu下anaconda以及pytorch cpu版本的安装过程，是笔者安装过后回忆的版本，可能有诸多不足之处，希望大家谅解。