查看 cuda 版本

1
nvidia-smi

或者右键选择 nvidia控制面板，点击 帮助 -> 系统信息 -> 组件

创建 conda 虚拟环境

1
2
3
4
# 创建虚拟环境
conda create --name 虚拟环境名称 python=3.9
# 激活虚拟环境
conda activate 虚拟环境名称

安装 pytorch

1
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121