python - 无法在 Ubuntu 中安装 Pytorch

Question

我正在使用以下命令在我的 conda 环境中安装 pytorch。

conda install pytorch=0.4.1 cuda90 -c pytorch

但是，我收到以下错误

解决环境：失败

PackagesNotFoundError：当前频道不提供以下软件包：

• 火炬=0.4.1
• cuda90

当前频道：

• https://conda.anaconda.org/pytorch/linux-32
• https://conda.anaconda.org/pytorch/noarch
• https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/linux-32
• https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/noarch
• https://repo.anaconda.com/pkgs/main/linux-32
• https://repo.anaconda.com/pkgs/main/noarch
• https://repo.anaconda.com/pkgs/free/linux-32

要搜索可能提供您正在寻找的 conda 包的替代频道，请导航至

https://anaconda.org

我该如何解决这个问题？我当然已经安装了 cuda 9 并且 nvcc 可以工作。

标签： pythonlinuxanacondapytorchconda

Answer 1

直接访问 pytorch 网站并按照设置说明进行操作，它会准确告诉您安装所需的命令 - pytorch - 开始

例如：

如果您正在寻找旧版本的 PyTorch，可以在此处找到版本历史记录和要安装的命令 -安装 PyTorch 的先前版本

如果这对您不起作用，您最后的选择是自己从源代码构建。这是 0.4.1 版的 GitHub 存储库 - pytorch at 0.4.1。从源代码安装的步骤在此处的 repo 中进行了概述。