PackagesNotFoundError：安装pytorch时cudatoolkit=11.1.0

Question

我从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

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

Answer 1

您需要-c conda-forge用于最新安装。来自官方网站的命令是：

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

https://pytorch.org

Answer 2

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

Answer 3

从https://pytorch.org/get-started/locally/，我得到的安装命令是

conda install pytorch torchvision torchaudio cudatoolkit=11.0 -c pytorch

cuda版本是11.0，而不是11.1.0。尝试以上方法。