【逆天大坑】pytorch官网方法安装居然是cpu版本 | 【解决方法】

pytorch官网给出了两种安装方法
conda pytorch, 去这里检查conda的pytorch的历史版本
pip torch， 去这里检查pip的torch的历史版本

pytorch和torch的区别

两者是一个东西。
pytorch是conda里的名字

装cpu版本：conda install pytorch cpuonly -c pytorch
装GPU版本：conda install pytorch pytorch-cuda=12.1 -c pytorch -c nvidia

torch是pip里的名字

装GPU版本：pip install torch
装cpu版本：pip install torch --index-url https://download.pytorch.org/whl/cpu

conda安装

会一直卡在solving environment

解决方法：分解安装

conda install pytorch -c pytorch -c nvidia
conda install torchvision -c pytorch -c nvidia
conda install torchaudio -c pytorch -c nvidia
conda install pytorch-cuda=12.1 -c pytorch -c nvidia

貌似成功了，测试一下：

啊？怎么输出是 False ？那不G了？
仔细观察，发现默认装的是pytorch的CPU版本

后来上conda官网一番推敲发现：

conda 要配合 pytorch-cuda 和 pytorch一起才能安装GPU版本，分开装不行！
so，再来一次：

conda install pytorch pytorch-cuda=12.1 -c pytorch -c nvidia
conda install torchvision pytorch-cuda=12.1 -c pytorch -c nvidia
conda install torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia

然后又一直卡在 solving environment
所以，我直接放弃了conda的安装方法。

pip 安装

1 在线安装

conda create -n env python=3.10
pip3 install torch torchvision torchaudio

这里等待了差不多一个小时的时间（网速均速 300kb/s ~ 400kb/s），所以我接下来会讲手动安装的方法

检验一下：

nice！一次就成功！

2 手动安装

见我的博客