英伟达显卡的环境安装

 

1. 安装CUDA
2. 安装pytorch GPU版本（需要有英伟达显卡，其他命令官网可查）
   1. conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch
3. 注意CUDA、显卡、显卡驱动、pytorch之间的版本兼容问题
   1. CUDA下载：CUDA Toolkit 11.5 Downloads | NVIDIA Developer
   2. 显卡驱动下载：NVIDIA 驱动程序下载
   3. Pytorch和CUDA版本匹配查询：Previous PyTorch Versions | PyTorch
   4. GPU是否支持CUDA：CUDA GPUs | NVIDIA Developer

pytorch的pip安装命令

CPU版本：

# Linux
pip install torch==1.9.0+cpu torchvision==0.10.0+cpu torchaudio==0.9.0 -f https://download.pytorch.org/whl/torch_stable.html
# Mac & Windows
pip install torch torchvision torchaudio

GPU版本：（安装包超过3g）

# Linux & Windows
pip install torch==1.9.0+cu111 torchvision==0.10.0+cu111 torchaudio==0.9.0 -f https://download.pytorch.org/whl/torch_stable.html

常用numpy，同时常用的数据结构为tensor；基本操作需要熟练掌握。