1、查看cuda版本 在命令行输入如下代码: nvidia-smi 本机CUDA版本为12.1 2、在官网选择对应的版本 网址为: https://pytorch.org/ 选择对应信息,获取到pytorch下载命令 在虚拟环境中输入命令: 注:请输入自己对应的命令 pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121 下载完成<