如何在WIn10+Anaconda環境中安裝PyTorch

這篇文章將為大家詳細講解有關如何在WIn10+Anaconda環境中安裝PyTorch，文章內容質量較高，因此小編分享給大家做個參考，希望大家閱讀完這篇文章后對相關知識有一定的了解。

1. 為 PyTorch 創建虛擬環境

關于 Anaconda 的安裝步驟這里就忽略不講了，Win10 下安裝 Anaconda 非常簡單。

安裝 Anaconda 完畢后，我們在安裝 PyTorch 之前最好先創建一個 pytorch 的虛擬環境。之所以創建虛擬環境是因為 Python 為不同的項目需求創建不同的虛擬環境非常常見。在實際項目開發中，我們通常會根據自己的需求去下載各種相應的框架庫，但是可能每個項目使用的框架庫并不一樣，或使用框架的版本不一樣，這樣需要我們根據需求不斷的更新或卸載相應的庫，管理起來相當麻煩。所以通過創建虛擬環境，相當于為不同的項目創建一塊獨立的空間，在這個空間里，你安裝任何庫和框架都是獨立的，不會影響到外部環境。

總之就是想象成為 PyTorch 單獨創建一個空間，這個空間與其它空間互不影響。我之前安裝了 TensorFlow 了，也是創建了一個虛擬環境，且 Python 版本是 3.5 的。所以，這里當然也要創建虛擬環境了。

創建 PyTorch 虛擬環境步驟：

1.打開 Anaconda 自帶的 Anaconda Prompt：

2.打開 Anaconda Prompt 之后，在命令行輸入：

conda create --name pytorch python=3.6

注意，這里的 pytorch 是虛擬環境的名稱，可隨意取。

3.然后繼續下列命令，進入 pytorch 虛擬環境：

activate pytorch

好了，pytorch 的虛擬環境創建好了。接下來就是安裝 PyTorch 了。

其實，安裝 PyTorch 還是非常簡單的。接下來記錄一些坑。

2. PyTorch 官網安裝

PyTorch 的官網提供了簡單的安裝方法，只需簡單的命令即可。

首先，打開 PyTorch 官網安裝頁面（需自備梯子）：https://pytorch.org/get-started/locally/

可以看到頁面里的這張圖，非常簡單。我們的環境是：Windows、Conda、Python 3.6、None CUDA，對應的命令是：

conda install pytorch-cpu torchvision-cpu -c pytorch

復制這條命令在 Anaconda Prompt 里輸入，則 PyTorch “順利”安裝完成。

但是問題來了，在 Jupyter 里 import torch 沒有問題，可 import torchvision 確報錯了：

ImportError: DLL load failed: 找不到指定的模塊。

可能是 pillow 模塊有問題。嘗試卸載 pillow 模塊，但是會同時卸載 PyTorch，故為一坑。

3. pip 安裝

上一種方法舍棄，這里嘗試使用 pip 安裝。

同樣在 Anaconda Prompt 里輸入以下命令：

# Python 3.6
pip3 install https://download.pytorch.org/whl/cpu/torch-1.0.0-cp36-cp36m-win_amd64.whl
pip3 install torchvision

安裝順利完成。

這時候在 Jupyter 里 import torch 沒有問題，可 import torchvision 確仍然報錯了：

ImportError: DLL load failed: 找不到指定的模塊。

解決方案：

在 Anaconda Prompt 里輸入以下命令：

conda uninstall pillow
pip install pillow

使用 pip 重新安裝 pillow 模塊即可。

關于如何在WIn10+Anaconda環境中安裝PyTorch就分享到這里了，希望以上內容可以對大家有一定的幫助，可以學到更多知識。如果覺得文章不錯，可以把它分享出去讓更多的人看到。