PyTorch 安装

PyTorch 是一个深度学习框架

安装

1. 安装环境

   • Python 版本: 3.9 或更高
   • Python venv 虚拟环境

   sudo apt install python3 pyhon3-venv

2. 配置虚拟环境

   # 创建代码目录
   mkdir learn
   cd learn
   
   # 创建并切换到虚拟环境
   python3 -m venv .venv
   source .venv/bin/activate

   出现 (.venv) 字样后说明已切换到虚拟环境。退出虚拟环境时需要执行以下命令：

   deactivate

3. 安装 PyTorch

   • CPU 版

     pip3 install torch

   • GPU 版

     见 PyTorch 官网 指引