PyTorch 根据官网命令行无法安装 GPU 版本 解决办法

最近遇到一个问题，PyTorch 官网给出了 GPU 版本的安装命令，但安装成功后查看版本，仍然是 torch 2.6.0+cpu

1. 清理现有 PyTorch 安装

经过探索发现，需要同时卸载 conda 和 pip 安装的 torch。

conda remove pytorch torchvision torchaudio --force
pip uninstall torch torchvision torchaudio -y

楼主先前只用 pip 安装，但用 conda 卸载后才发现 conda 也偷偷安装了。 

2. 升级 pip 工具

清理 pip 缓存；升级，确保 pip 工具能够解析 --index-url:

pip cache purge
pip install --upgrade pip

 3. 强制不使用缓存安装

使用清华源 https://pypi.tuna.tsinghua.edu.cn/simple

pip3 install torch torchvision torchaudio --no-cache-dir --index-url https://download.pytorch.org/whl/cu126 -i https://pypi.tuna.tsinghua.edu.cn/simple

cu 后改成自己 cuda 的版本， 比如我的 cuda 是12.6，就用 cu126；如果 cuda 12.4，就用 cu124

nvcc --version

4. 验证安装 

可以通过如下命令验证：

python -c "import torch; print(torch.__version__, torch.cuda.is_available(), torch.version.cuda)"

应该输出：2.6.0 True 12.6 <Your GPU Name>

至此就能大功告成啦！