它不仅功能强大,而且易于上手,尤其适合研究和开发
然而,对于许多初学者来说,在Linux系统下安装PyTorch可能是一项挑战
本文将详细介绍如何在Linux系统下高效、稳定地安装PyTorch,帮助大家快速搭建深度学习环境
一、准备工作 在开始安装之前,确保你的系统满足以下基本要求: - 操作系统:Linux(推荐使用Ubuntu或其他主流发行版)
- Python版本:建议使用Python 3.7或更高版本
Python是PyTorch的基石,合适的版本能够确保兼容性和性能
- 硬件支持:根据你的需求,确定是否需要GPU支持
如果计算机配备了NVIDIA GPU,并且希望加速深度学习计算,可以安装GPU版本的PyTorch
此外,建议使用虚拟环境来管理项目依赖,以避免包冲突
Anaconda是一个流行的虚拟环境管理工具,它不仅能方便地创建和管理虚拟环境,还能提供大量的科学计算包
二、安装Anaconda 1.下载Anaconda 访问Anaconda官网,根据你的系统选择对应的版本
Anaconda会自动检测你的操作系统,并推荐合适的版本
你也可以选择从国内镜像站(如清华大学开源软件镜像站)下载,以提高下载速度
使用以下命令在Linux终端下载Anaconda: bash wget https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/Anaconda3-2023.03-Linux-x86_64.sh 2.安装Anaconda 下载完成后,执行以下命令进行安装: bash bash Anaconda3-2023.03-Linux-x86_64.sh 阅读并接受安装协议,确认安装位置(默认即可),等待安装完成
3.配置环境变量 安装完成后,需要配置环境变量以便系统能够正确找到Anaconda
编辑`~/.bashrc`文件,添加以下内容: bash export PATH=/root/anaconda3/bin:$PATH 这里的路径根据实际情况调整 source ~/.bashrc 4.初始化Anaconda 执行以下命令,使conda的基础环境在启动时不被激活: bash conda init conda config --setauto_activate_base false 5.配置conda源 为了提高包下载和更新的速度,建议使用国内镜像源
例如,使用清华镜像源: bash conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/ conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/ conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/ conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/ conda config --setshow_channel_urls yes 三、创建虚拟环境 使用Anaconda创建虚拟环境可以隔离不同项目的依赖项,避免项目之间的冲突
1.创建虚拟环境 执行以下命令创建一个新的虚拟环境,例如名为`pytorch`的环境,并指定Python版本为3.8: bash co