WIn10+Anaconda 環境下安裝 PyTorch 避坑指南

紅色石頭發表於2019-01-28

這些天安裝 PyTorch,遇到了一些坑,特此總結一下,以免忘記。分享給大家。

首先,安裝環境是:作業系統 Win10,已經預先暗轉了 Anaconda。

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 模組即可。

至此,填坑。PyTorch 成功安裝。


相關文章