温馨提示×

Ubuntu PyTorch如何进行环境配置

小樊
45
2025-09-03 01:16:15
栏目: 智能运维

以下是在Ubuntu上配置PyTorch环境的步骤:

  1. 更新系统sudo apt updatesudo apt upgrade
  2. 安装Python和pipsudo apt install python3 python3-pip
  3. 创建虚拟环境(可选)python3 -m venv pytorch-envsource pytorch-env/bin/activate
  4. 安装CUDA和cuDNN(可选,用于GPU加速)
    • 安装CUDA:下载对应版本CUDA Toolkit,如wget https://developer.download.nvidia.com/compute/cuda/11.7.1/local_installers/cuda-repo-ubuntu2004-11-7-local_11.7.1-450.51.06-1_amd64.deb,然后按步骤安装。
    • 安装cuDNN:下载与CUDA版本匹配的cuDNN库,解压后复制文件到指定目录。
  5. 安装PyTorch
    • pip安装:根据CUDA版本选择命令,如CPU版本pip install torch torchvision torchaudio,CUDA 11.7版本pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu117
    • conda安装:先安装Miniconda,创建环境后,如conda install pytorch torchvision torchaudio cudatoolkit=11.7 -c pytorch -c conda-forge
  6. 验证安装:在Python中运行import torchprint(torch.__version__)print(torch.cuda.is_available())

0