1)创建并激活环境
conda create -n DOT1 python==3.9
conda activate DOT1
2)方法1:命令行安装torch
① 查看cuda版本【12.6】
nvidia-smi # 查看电脑支持的cuda版本
nvcc -V # 查看安装的cuda版本
② 去pytorch官网找命令
pytorch官网PyTorch
③ 直接输入图中命令:
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu126
3)方法2:本地安装
① 输入命令
pip3 install torch