Pytorch 环境搭建

Pytorch 环境搭建

环境配置:学习 Anaconda3 教程,然后配置 PyTorch。

本地服务器支持 GPU

  1. 替换 本地服务器 GPU 配置 中的环境 aitorch

  2. 安装 PyTorch GPU 支持:

    conda install pytorch torchvision torchaudio -c pytorch
    
  3. 添加 Jupyter 内核支持:

    conda install ipykernel
    python -m ipykernel install --user --name torch --display-name "PyTorch"
    
  4. 更多内容见 预览pytorch-demo