• ???? 声明: 作为全网 AI 领域 干货最多的博主之一,❤️ 不负光阴不负卿 ❤️

Linux下 pytorch 1.8的极简安装_服务器


???? Linux下 pytorch 1.8的极简安装


进入 pytorch 主页

  • ???? ​​pytorch.org/​​

根据你的机型进行选择

Linux下 pytorch 1.8的极简安装_深度学习_02

copy 给出的命令 到 服务器 shell 命令窗口,执行即可

pip3 install torch==1.8.1+cu111 torchvision==0.9.1+cu111 torchaudio==0.8.1 -f https://download.pytorch.org/whl/lts/1.8/torch_lts.html

????这里涉及到一个 conda 和 pip 下载方式的问题


  • 首先,不论哪种 方式,都是可以的,但是都建议 配置 源 进行 下载加速
  • 我自己平时 习惯会首先使用 pip 来进行 安装
  • 推荐 conda 和 pip 的 加速配置方式如下
  • conda 主要功能是在服务器中 搭建自己独立环境【不影响其他人】
  • ???? anaconda conda 切换为国内源 |windows 和 Linux配置方法
  • ???? linux和window设置 pip 镜像源 | 最实用的环境下载加速设置

????其它版本 pytorch 的安装


打开这个页面,下方均有对应的 conda 和 pip 安装 命令 结合 自己 电脑 cuda 版本进行选择即可

  • ???? ​​pytorch.org/get-started…​​

Linux下 pytorch 1.8的极简安装_人工智能_03


高版本 Cuda 向下兼容低版本


举例:服务器安装了高版本 驱动的内核, 例如 11.2 版本的 CUDA 及 内核

那么,目前,你的 torch 版本 就可以随意安装;

但是如果服务器 驱动版本 较低,只支持 CUDA 10.x 版本,这时就 必须要 选择 9.X 和 10.X 匹配 的 torch 版本进行安装;


墨理学AI


  • 作为全网 AI 领域 干货最多的博主之一,❤️ 不负光阴不负卿 ❤️
  • ❤️ 如果文章对你有帮助、点赞、评论鼓励博主的每一分认真创作

快乐学AI 、深度学习环境搭建 : 一文读懂

  • ???? # ubuntu给当前用户安装cuda11.2 图文教程
  • ???? # linux和window设置 pip 镜像源——最实用的机器学习库下载加速设置
  • ???? # 指定GPU运行和训练python程序 、深度学习单卡、多卡 训练GPU设置【一文读懂】
  • ???? # Linux下cuda10.0安装Pytorch和Torchvision【一文读懂】
  • ???? # 一文读懂SSH密码登录、公钥认证登录
  • ???? # 一文读懂 Centos、Ubuntu 环境 安装JDK 11:配置JAVA_HOME环境变量

Linux下 pytorch 1.8的极简安装_linux_04