WIn10+Anaconda 环境下安装 PyTorch 避坑指南
紅色石頭的個人網站:redstonewill.com
這些天安裝 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 之后,在命令行輸入:
注意,這里的 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,對應的命令是:
復制這條命令在 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 成功安裝。
與50位技術專家面對面20年技術見證,附贈技術全景圖
總結
以上是生活随笔為你收集整理的WIn10+Anaconda 环境下安装 PyTorch 避坑指南的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: 为什么铺天盖地都是Python的广告,它
- 下一篇: 7年专科生程序员同时去腾讯和微软面试,问