無法在 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 package 的備用頻道，請導航至

https://anaconda.org

我該如何解決這個問題？ 我當然安裝了 cuda 9 並且 nvcc 工作。

Answer 1

Go 直接訪問 pytorch 網站並按照設置說明進行操作，它會准確告訴您安裝所需的命令 - pytorch - 開始

例如：

如果您正在尋找舊版本的 PyTorch，可以在此處找到版本歷史記錄和要安裝的命令 -安裝 PyTorch 的先前版本

如果這對您不起作用，您最后的選擇是自己從源代碼構建。 這是版本 0.4.1 的 GitHub 存儲庫 - pytorch 在 0.4.1 。 從源代碼安裝的步驟在此處的 repo 中進行了概述。

Answer 2

在我看來，這是給出問題的版本號。 對於初學者，要指定特定版本，您不能使用“=”符號。 如果您特別尋找您指定的相同版本，我建議您嘗試使用 pytorch0.4.1 作為 package 名稱，否則您可以通過以下方式獲取最新版本：

conda install -c pytorch pytorch

您可以參考此站點的 CUDA： https://anaconda.org/anaconda/cudatoolkit

或者如果這適合您，請運行此命令

pip3 install pycuda