WIn10+Anaconda環(huán)境下安裝PyTorch(避坑指南)
這些天安裝 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)境是因?yàn)?Python 為不同的項(xiàng)目需求創(chuàng)建不同的虛擬環(huán)境非常常見。在實(shí)際項(xiàng)目開發(fā)中,我們通常會根據(jù)自己的需求去下載各種相應(yīng)的框架庫,但是可能每個項(xiàng)目使用的框架庫并不一樣,或使用框架的版本不一樣,這樣需要我們根據(jù)需求不斷的更新或卸載相應(yīng)的庫,管理起來相當(dāng)麻煩。所以通過創(chuàng)建虛擬環(huán)境,相當(dāng)于為不同的項(xiàng)目創(chuàng)建一塊獨(dú)立的空間,在這個空間里,你安裝任何庫和框架都是獨(dú)立的,不會影響到外部環(huán)境。
總之就是想象成為 PyTorch 單獨(dú)創(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ù)下列命令,進(jìn)入 pytorch 虛擬環(huán)境:
activate pytorch

好了,pytorch 的虛擬環(huán)境創(chuàng)建好了。接下來就是安裝 PyTorch 了。
其實(shí),安裝 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 確報(bào)錯了:
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 確仍然報(bào)錯了:
ImportError: DLL load failed: 找不到指定的模塊。
解決方案:
在 Anaconda Prompt 里輸入以下命令:
conda uninstall pillow pip install pillow
使用 pip 重新安裝 pillow 模塊即可。
至此,填坑。PyTorch 成功安裝。以上就是本文的全部內(nèi)容,希望對大家的學(xué)習(xí)有所幫助,也希望大家多多支持腳本之家。
相關(guān)文章
Pytorch mask_select 函數(shù)的用法詳解
今天小編就為大家分享一篇Pytorch mask_select 函數(shù)的用法詳解,具有很好的參考價值,希望對大家有所幫助。一起跟隨小編過來看看吧2020-02-02
詳解Python中生成隨機(jī)數(shù)據(jù)的示例詳解
在日常工作編程中存在著各種隨機(jī)事件,同樣在編程中生成隨機(jī)數(shù)字的時候也是一樣。每當(dāng)在?Python?中生成隨機(jī)數(shù)據(jù)、字符串或數(shù)字時,最好至少大致了解這些數(shù)據(jù)是如何生成的。所以本文將詳細(xì)為大家講解一下Python是如何生成隨機(jī)數(shù)據(jù),需要的可以參考一下2022-04-04
python函數(shù)遞歸調(diào)用的實(shí)現(xiàn)
本文主要介紹了python函數(shù)遞歸調(diào)用的實(shí)現(xiàn),文中通過示例代碼介紹的非常詳細(xì),對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值,需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2023-05-05
django多文件上傳,form提交,多對多外鍵保存的實(shí)例
今天小編就為大家分享一篇django多文件上傳,form提交,多對多外鍵保存的實(shí)例,具有很好的參考價值,希望對大家有所幫助。一起跟隨小編過來看看吧2019-08-08
pytest實(shí)現(xiàn)多種調(diào)用方式
pytest是一個非常成熟的全功能的Python測試框架,本文主要介紹了pytest多種調(diào)用方式,具有一定的參考價值,感興趣的可以了解一下2023-12-12
matplotlib繪制兩點(diǎn)間連線的幾種方法實(shí)現(xiàn)
本文主要介紹了matplotlib繪制兩點(diǎn)間連線的幾種方法實(shí)現(xiàn),主要介紹了4種方法,文中通過示例代碼介紹的非常詳細(xì),具有一定的參考價值,感興趣的小伙伴們可以參考一下2022-03-03

