基本应用软件
本文使用 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())