PyTorch 深度学习环境准备

278 阅读1分钟

基本应用软件

本文使用 Anaconda 管理 Python 发行版,避免操作系统自带 Python 版本干扰。

for Windows

按照顺序安装:

  • Windows Terminal(Windows 11 自带不用装,Windows 10 从商店安装)
  • Powershell 7(可以从商店安装,也可以用 winget)
  • Anaconda
  • Visual Studio Code(可选,作为 Git 的默认编辑器)
  • Git
  • PyCharm

Terminal 中增加 Anaconda Profile

先在「所有应用」中找到「Anaconda Powershell Prompt (anaconda3)」,右键打开所在位置,右键打开属性,复制从 -ExecutionPolicy 开始的部分。

添加新配置文件,复制自 Powershell,名称改为 Anaconda,在 pwsh.exe" 后面粘贴刚才复制的内容,图标改为 %USERPROFILE%\anaconda3\Menu\anaconda-navigator.ico,保存。以后就可以在 Terminal 里直接打开 Anaconda 环境了。

for MacOS

安装:

  • iTerm2
  • Miniconda
  • PyCharm

Anaconda 配置

现在 Anaconda 在国内有了加速,而且国内比如清华的源实测速度非常慢,所以不建议换源。另外,conda-forge 频道的包可能会对 PyTorch 的包有负面影响,也不建议增加。所以,Anaconda 尽量保持默认配置不变。网上其他帖子说的换源等操作都已经过时。

更新 conda,以支持最新版本 Python 包:

conda update conda
conda update anaconda

PyTorch 虚拟环境

创建虚拟环境,-n 后面是自己定义的环境名:

conda create -n qfni python=3.10
conda activate qfni

请注意,可能有的教程会让你安装 nVidia CUDA 相关的

按照 PyTorch 官网 给的命令,安装 PyTorch:

# Windows 有 nVidia GPU
conda install pytorch torchvision torchaudio pytorch-cuda=11.7 -c pytorch -c nvidia

# Windows 没有 nVidia GPU,以及 MacOS
conda install pytorch torchvision torchaudio -c pytorch

先执行 python,然后测试一下:

import torch
x = torch.rand(5, 3)
print(x)

针对 nVidia GPU,测试 CUDA:

import torch
torch.cuda.is_available()

针对 Apple Siliccon,测试 MPS:

import torch
print(torch.backends.mps.is_available())