這些天安裝 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 成功安裝。