PackagesNotFoundError: cudatoolkit=11.1.0 when installing pytorch

PackagesNotFoundError: cudatoolkit=11.1.0 when installing pytorch

我从 https://pytorch.org/get-started/locally/ 获取安装命令。当我 运行 命令时: conda install pytorch torchvision torchaudio cudatoolkit=11.1.0 -c pytorch 我收到以下错误:

Collecting package metadata (current_repodata.json): done
Solving environment: failed with initial frozen solve. Retrying with flexible solve.
Collecting package metadata (repodata.json): done
Solving environment: failed with initial frozen solve. Retrying with flexible solve.

PackagesNotFoundError: The following packages are not available from current channels:

  - cudatoolkit=11.1.0

Current channels: like

  - https://conda.anaconda.org/pytorch/linux-64
  - https://conda.anaconda.org/pytorch/noarch
  - https://repo.anaconda.com/pkgs/main/linux-64
  - https://repo.anaconda.com/pkgs/main/noarch
  - https://repo.anaconda.com/pkgs/r/linux-64
  - https://repo.anaconda.com/pkgs/r/noarch

To search for alternate channels that may provide the conda package you're
looking for, navigate to

    https://anaconda.org

and use the search bar at the top of the page.

然后我一一尝试了以下命令:

(base) eslam@scholar:~$ conda config --append channels conda-forge
(base) eslam@scholar:~$ install cudatoolkit=11.1.0
install: missing destination file operand after 'cudatoolkit=11.1.0'
Try 'install --help' for more information.
(base) eslam@scholar:~$ conda install -c conda-forge cudatoolkit=11.1.0

获得与上面相同的输出。

(base) eslam@scholar:~$ conda info --envs
# conda environments:
#
base                  *  /home/eslam/anaconda3

请问我该如何解决这个问题?

来自https://pytorch.org/get-started/locally/,我得到的安装命令是

conda install pytorch torchvision torchaudio cudatoolkit=11.0 -c pytorch

cuda 版本是 11.0,不是 11.1.0。试试上面的方法。

您需要使用 -c conda-forge 进行最新安装。来自官方网站的命令是:

conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch -c conda-forge

https://pytorch.org

您可以 search cudatoolkit in anaconda.org 并找到包含您的架构和版本的频道。就我而言,我在 Windows 下安装 PyTorch 1.8.1,但显然 PyTorch 首页上提到的 nvidia 频道不包含 Windows 的任何软件包,所以我不得不切换到 conda-forge,其中可以找到 cudatoolkit 11.1(例如,从上面的搜索结果)。

这对我有用:conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch -c nvidia -c conda-forge