Ubuntu安装pytorch

系统环境

Ubuntu 18.04.3,查看命令

安装过程

由于不是安装官网最新的Pytorch版(当时最新1.7,安装1.5),所以选择anaconda进行安装,它自动管理依赖包之间关系,自动下载符合依赖关系的包,不用自己去匹配包与包之间的版本对应关系。

安装anaconda3

  1. 前往清华镜像源选择适合自己系统的anaconda包。想下载最新版前往官网即可,但下载速度可能较慢。
  2. 复制相应的包链接,进行下载。如果linux系统无法访问外网,也可先下载到本地,然后通过xftp等工具上传到linux系统:
wget https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/Anaconda3-5.3.1-Linux-x86_64.sh
  1. 执行安装,运行前一步下载的.sh文件即可
bash Anaconda3-5.3.1-Linux-x86_64.sh
  1. 安装过程中会有一个输入自定义安装路径的过程,不输入就是默认路径,这个根据自己情况而定(笔者安装位置为/data/anaconda3),其余过程一直选yes即可,不过关于是否安装vscode时,选no(如果你不需要的话)
  2. 安装完成之后,在命令行输入下面命令,来验证是否安装成功
#命令
/data/anaconda3/bin/conda

#安装成功会输出下述内容:
usage: conda [-h] [-V] command ...

conda is a tool for managing and deploying applications, environments and packages.

Options:

positional arguments:
.....
  1. 考虑到此系统会被多个人同时使用,且未分配多个账户(共用一个root账号),所以在此未配置环境变量。从而每次输入的相关命令都需要具体路径

注:

  1. 此处及后面一直使用的anaconda的base环境。
  2. 运用conda命令安装包时,如若下载速度较慢,可尝试修改国内镜像源
  3. 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
...

运行如果不报错,表示安装成功

python包安装依赖版本关系查询

  • 0
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值