温馨提示×

PyTorch在Linux上如何配置

小樊
39
2025-08-15 17:06:07
栏目: 智能运维

以下是在Linux上配置PyTorch的步骤:

  1. 更新系统包sudo apt update && sudo apt upgrade -y
  2. 安装Python和pip:若未安装,使用sudo apt install python3 python3-pip
  3. 创建虚拟环境(可选):使用python3 -m venv pytorch_env创建,source pytorch_env/bin/activate激活。
  4. 安装PyTorch
    • pip安装:访问PyTorch官网获取命令,如CPU版本pip install torch torchvision torchaudio,CUDA版本如pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu117
    • conda安装:先安装Miniconda,再创建环境,然后conda install pytorch torchvision torchaudio cudatoolkit=版本号 -c pytorch(有GPU时指定CUDA版本,无GPU用cpuonly)。
  5. 验证安装:在Python环境中运行import torchprint(torch.__version__)print(torch.cuda.is_available()),查看版本和CUDA是否可用。

0