Debian上PyTorch的常用命令有哪些

以下是Debian上PyTorch的常用命令，涵盖安装、验证及环境管理：

1. 系统与依赖准备

• 更新系统包：
  sudo apt update && sudo apt upgrade -y
• 安装Python和pip：
  sudo apt install python3 python3-pip

2. PyTorch安装命令

• CPU版本（无GPU）：
  pip install torch torchvision torchaudio
• GPU版本（需先安装CUDA）：
  pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu117
  （cu117替换为实际CUDA版本，如cu102、cu120等）
• Conda安装（推荐隔离环境）：

  conda create -n pytorch_env python=3.9 -y  
  conda activate pytorch_env  
  conda install pytorch torchvision torchaudio cudatoolkit=11.7 -c pytorch  # 替换CUDA版本  
  

3. 验证安装

• 检查版本及CUDA可用性：
  python3 -c "import torch; print(torch.__version__); print(torch.cuda.is_available())"

4. 环境管理（可选）

• 创建虚拟环境（隔离项目）：
  python3 -m venv pytorch_env
source pytorch_env/bin/activate
• 退出虚拟环境：
  deactivate

5. 其他常用操作

• 卸载PyTorch：
  pip uninstall torch torchvision torchaudio
• 查看已安装包：
  pip list

说明：安装时需根据系统CUDA版本选择对应命令，可通过nvcc --version查看CUDA版本。推荐使用虚拟环境隔离不同项目依赖。