系统环境
Ubuntu 18.04.3,查看命令
安装过程
由于不是安装官网最新的Pytorch版(当时最新1.7,安装1.5),所以选择anaconda进行安装,它自动管理依赖包之间关系,自动下载符合依赖关系的包,不用自己去匹配包与包之间的版本对应关系。
安装anaconda3
- 前往清华镜像源选择适合自己系统的anaconda包。想下载最新版前往官网即可,但下载速度可能较慢。
- 复制相应的包链接,进行下载。如果linux系统无法访问外网,也可先下载到本地,然后通过xftp等工具上传到linux系统:
wget https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/Anaconda3-5.3.1-Linux-x86_64.sh
- 执行安装,运行前一步下载的.sh文件即可
bash Anaconda3-5.3.1-Linux-x86_64.sh
- 安装过程中会有一个输入自定义安装路径的过程,不输入就是默认路径,这个根据自己情况而定(笔者安装位置为/data/anaconda3),其余过程一直选yes即可,不过关于是否安装vscode时,选no(如果你不需要的话)
- 安装完成之后,在命令行输入下面命令,来验证是否安装成功
#命令
/data/anaconda3/bin/conda
#安装成功会输出下述内容:
usage: conda [-h] [-V] command ...
conda is a tool for managing and deploying applications, environments and packages.
Options:
positional arguments:
.....
- 考虑到此系统会被多个人同时使用,且未分配多个账户(共用一个root账号),所以在此未配置环境变量。从而每次输入的相关命令都需要具体路径。
注:
- 此处及后面一直使用的anaconda的base环境。
- 运用conda命令安装包时,如若下载速度较慢,可尝试修改国内镜像源
- anaconda自带Python环境,安装好anaconda后,即安装好了python环境。不同版本anaconda自带不同python版本,如不是自己所需python版本,可通过anaconda创建一个虚拟环境,并指定python版本即可,然后切换到对应的虚拟环境中操作即可。
安装pytorch GPU版
官网最新版安装命令
安装pytorch1.5 + torchtext + torchvision
cudatoolkit版本根据自己系统情况而定
/data/anaconda3/bin/conda install pytorch==1.5.0 torchvision torchaudio cudatoolkit=10.1 -c pytorch
验证
输入python命令进入交互模式
/data/anaconda3/bin/python
依次导包
import torch
import torchtext
import torcvision
...
运行如果不报错,表示安装成功