Debian系统中PyTorch安装步骤
1. 更新系统包
首先,确保Debian系统处于最新状态,避免因包版本冲突导致安装失败:
sudo apt update && sudo apt upgrade -y
2. 安装Python环境与pip
PyTorch依赖Python 3.6及以上版本,需安装Python3及pip(Python包管理工具):
sudo apt install python3 python3-pip -y
3. (可选)创建虚拟环境
为隔离项目依赖,避免与系统Python环境冲突,建议使用venv创建虚拟环境:
python3 -m venv pytorch_env # 创建名为"pytorch_env"的虚拟环境
source pytorch_env/bin/activate # 激活虚拟环境(激活后终端提示符会显示环境名)
4. 安装PyTorch
PyTorch提供CPU版本(无GPU加速)和GPU版本(需NVIDIA显卡+CUDA支持),需根据硬件配置选择:
① CPU版本安装(无GPU)
直接通过pip安装PyTorch核心包(torch、torchvision、torchaudio):
pip install torch torchvision torchaudio
此版本无需额外配置CUDA,适用于普通CPU计算场景。
② GPU版本安装(需NVIDIA显卡)
需先确认显卡型号及已安装的CUDA版本(可通过nvidia-smi命令查看CUDA版本),再选择对应PyTorch安装命令。例如:
- 若CUDA版本为11.7,使用以下命令:
pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu117 - 若CUDA版本为12.0,使用以下命令:
pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu120注:PyTorch官网会定期更新支持的CUDA版本,建议安装前访问PyTorch官方安装页面获取最新命令。
5. (可选)使用conda安装(替代方案)
若已安装Miniconda/Anaconda,可通过conda更便捷地管理PyTorch及CUDA工具包:
# 安装Miniconda(若未安装)
wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh
bash Miniconda3-latest-Linux-x86_64.sh # 按提示完成安装
# 创建conda环境(Python 3.9版本)
conda create -n pytorch_env python=3.9 -y
conda activate pytorch_env # 激活环境
# 安装PyTorch(以CUDA 11.7为例)
conda install pytorch torchvision torchaudio cudatoolkit=11.7 -c pytorch
conda会自动处理CUDA工具包的依赖关系,适合新手快速部署。
6. 验证安装
安装完成后,通过Python脚本验证PyTorch是否安装成功及CUDA是否可用:
python3 -c "
import torch
print('PyTorch版本:', torch.__version__)
print('CUDA是否可用:', torch.cuda.is_available()) # 若为True则表示GPU加速可用
"
若输出显示PyTorch版本号且torch.cuda.is_available()返回True(GPU版本),则说明安装成功。
注意事项
- CUDA驱动要求:GPU版本需提前安装NVIDIA官方驱动(版本≥450),可通过
nvidia-smi命令检查驱动版本; - 依赖项问题:若安装过程中出现依赖错误,可尝试升级pip(
pip install --upgrade pip)或手动安装缺失的依赖(如libopenblas-dev、liblapack-dev等); - 环境管理:建议始终在虚拟环境中安装PyTorch,避免污染系统Python环境。
以上就是关于“Debian系统中PyTorch安装步骤”的相关介绍,筋斗云是国内较早的云主机应用的服务商,拥有10余年行业经验,提供丰富的云服务器、租用服务器等相关产品服务。云服务器资源弹性伸缩,主机vCPU、内存性能强悍、超高I/O速度、故障秒级恢复;电子化备案,提交快速,专业团队7×24小时服务支持!
简单好用、高性价比云服务器租用链接:https://www.jindouyun.cn/product/cvm