conda默认下载cpu版的pytorch

问题描述

当我使用conda install pytorch安装pytorch时，总是给我默认安装cpu版本。如下所示，搜索到的pytorch是cpu_py39hdc00b08_0。

  Package                Version  Build               Channel          Size
─────────────────────────────────────────────────────────────────────────────
....
  + pytorch                2.2.0  cpu_py39hdc00b08_0  pkgs/main        74MB

也可以在安装完毕后，运行下面的代码：

import torch
torch.cuda.is_available()

如果得到True，说明是gpu版本；如果得到False说明是cpu版本。

尝试解决

网上的解决方法有，

1. 取消使用清华源

清华源中只有cpu版本的pytorch，改用默认源。我原本的确使用清华源，但是我移除所有清华源并清除cache缓存后，使用默认源得到的仍然是cpu版本。