温馨提示×

Ubuntu上PyTorch资源如何获取

小樊
42
2025-08-22 00:26:18
栏目: 智能运维

在Ubuntu上获取PyTorch资源,可参考以下方法:

  • 安装PyTorch库
    • 使用Anaconda安装:先创建虚拟环境,如conda create -n pytorch_env python3.9,激活后conda install pytorch torchvision torchaudio pytorch-cuda11.8 -c pytorch -c nvidia(将11.8替换为所需CUDA版本)。
    • 使用pip安装:通用版本可pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118,指定版本安装则修改对应版本号。
  • 安装依赖库:若需GPU加速,要安装NVIDIA驱动、CUDA工具包和cuDNN库。可先通过nvidia - smi查看显卡及CUDA版本,再从NVIDIA官网下载对应版本的安装包进行安装。

0