Linux下 pytorch 1.8的极简安装

211 阅读2分钟

「这是我参与2022首次更文挑战的第30天,活动详情查看:2022首次更文挑战

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


    💜 Linux下 pytorch 1.8的极简安装


    进入 pytorch 主页

    根据你的机型进行选择

    1-0

    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 下载方式的问题



    🎄其它版本 pytorch 的安装


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

    2-0


    🎄高版本 Cuda 向下兼容低版本


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

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

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


    🚀🚀 墨理学AI


  • 🎉 作为全网 AI 领域 干货最多的博主之一,❤️ 不负光阴不负卿 ❤️
  • ❤️ 如果文章对你有帮助、点赞、评论鼓励博主的每一分认真创作
  • 快乐学AI 、深度学习环境搭建 : 一文读懂

    765_blue.jpg