手动安装GPU版本PyTorch

创建conda虚拟环境

conda create-ndlpython=3.9conda activate dl

手动下载轮子

👉 https://download.pytorch.org/whl/cu121/
说明：
cu121 = CUDA 12.1（根据你驱动版本匹配）

进去后找这三个（按你 Python 版本选）：
如果是 Python 3.9：

torch-2.5.1+cu121-cp39-cp39-win_amd64.whl torchvision-0.20.1+cu121-cp39-cp39-win_amd64.whl torchaudio-2.5.1+cu121-cp39-cp39-win_amd64.whl

本地安装

进入下载目录：

pipinstalltorch-*.whl torchvision-*.whl torchaudio-*.whl

验证

importtorch print(torch.cuda.is_available())print(torch.cuda.get_device_name(0))

看到True就成功了！！！