這些天安裝 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
新聞熱點
疑難解答