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

首先，安裝環(huán)境是：操作系統(tǒng) Win10，已經(jīng)預(yù)先暗轉(zhuǎn)了 Anaconda。

1. 為 PyTorch 創(chuàng)建虛擬環(huán)境

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

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

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

創(chuàng)建 PyTorch 虛擬環(huán)境步驟：

1.打開 Anaconda 自帶的 Anaconda Prompt：

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

conda create --name pytorch python=3.6

注意，這里的 pytorch 是虛擬環(huán)境的名稱，可隨意取。

3.然后繼續(xù)下列命令，進入 pytorch 虛擬環(huán)境：

activate pytorch

好了，pytorch 的虛擬環(huán)境創(chuàng)建好了。接下來就是安裝 PyTorch 了。

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

2. PyTorch 官網(wǎng)安裝

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

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

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

conda install pytorch-cpu torchvision-cpu -c pytorch

復(fù)制這條命令在 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 成功安裝。以上就是本文的全部內(nèi)容，希望對大家的學(xué)習(xí)有所幫助，也希望大家多多支持腳本之家。

最新評論