「这是我参与2022首次更文挑战的第30天,活动详情查看:2022首次更文挑战」
🎉 声明: 作为全网 AI 领域 干货最多的博主之一,❤️ 不负光阴不负卿 ❤️
💜 Linux下 pytorch 1.8的极简安装
进入 pytorch 主页
根据你的机型进行选择
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 主要功能是在服务器中 搭建自己独立环境【不影响其他人】
🎄其它版本 pytorch 的安装
打开这个页面,下方均有对应的 conda 和 pip 安装 命令 结合 自己 电脑 cuda 版本进行选择即可
🎄高版本 Cuda 向下兼容低版本
举例:服务器安装了高版本 驱动的内核, 例如 11.2 版本的 CUDA 及 内核
那么,目前,你的 torch 版本 就可以随意安装;
但是如果服务器 驱动版本 较低,只支持 CUDA 10.x 版本,这时就 必须要 选择 9.X 和 10.X 匹配 的 torch 版本进行安装;
🚀🚀 墨理学AI
🎉 作为全网 AI 领域 干货最多的博主之一,❤️ 不负光阴不负卿 ❤️ ❤️ 如果文章对你有帮助、点赞、评论鼓励博主的每一分认真创作
快乐学AI 、深度学习环境搭建 : 一文读懂