如何在Linux上安装PyTorch？

AI是当下最热门的话题之一。而且大部分的AI相关框架都是使用python进行开发。正所谓万事开头难，第一步当然是搭建python开发环境。

安装

• 安装anaconda: sudo pacman -Sy anaconda

  Please run
      $ source /opt/anaconda/bin/activate root
      $ source /opt/anaconda/bin/deactivate root
  to activate and deactivate the anaconda enviroment.
  

• 进入anaconda shell：source /opt/anaconda/bin/activate root

• 创建虚拟环境: conda create -n pytorch python=3.11

  # To activate this environment, use
  #     $ conda activate pytorch
  #
  # To deactivate an active environment, use
  #     $ conda deactivate
  

• 启用: conda activate pytorch

• 查看安装包: pip list

• 查看cuda版本：nvidia-smi

• 下载PyTorch，并按照步骤安装PyTorch

• 检查cuda是否可用

  import torch
  torch.cuda.is_available()
  

参考

• Install PyTorch tutorial