利用 Conda 安裝深度學習框架 Pytorch

1. 前言

Pytorch 目前是煊赫一時的深度學習框架。和 TensorFlow 比較起來學習曲線更加平滑，不用寫大量的樣板代碼就能夠對網絡進行訓練和使用。在最新版本的 Pytorch 中開始支持 Java 。可是安裝 Pytorch 並非很容易的事。今天就來講一下如何利用 Conda 安裝 Pytorch 。

2. Conda

這裏簡單提一下 Conda ，Conda 是一個開源的、跨平臺的軟件包管理系統和環境管理系統，用於安裝多個 Python 版本的軟件包及其依賴關係，並在它們之間輕鬆切換。你能夠把 Conda 看做 Python 領域的 Maven ，固然有些功能可能比 Maven 更增強大。一般我會安裝 anaconda 來集成 Conda 和 Python 環境，anacoda 額外還有 numpy 、pandas 等有用的科學計算包。若是你有潔癖也可使用 MiniConda ，它只包含 Conda 和 Python 。若是你還不會 Conda 建議你花十分鐘快速入門一下。

3. 在線安裝Pytorch

以 Mac 爲例，Conda 安裝最新版 Pytorch 的命令以下：

conda install pytorch torchvision -c pytorch
複製代碼

根據該命令的提示很容易就能安裝好 Pytorch。若是安裝失敗，極可能是由於網絡的延遲太高而超時。因此咱們能夠經過配置 anacoda 國內鏡像源來加速安裝。命令執行如下命令：

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --set show_channel_urls yes
複製代碼

而後從新安裝 Pytorch 便可。

https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/ 爲 Pytorch 的鏡像源，咱們在下一個章節也會用到它。

4. 離線安裝Pytorch

有時候即便使用鏡像源也會失敗，因此咱們能夠經過離線安裝來解決這個問題。當 Pytorch 安裝失敗後命令行終端若是出現下列提示：

CondaHTTPError: HTTP 000 CONNECTION FAILED for url <https://conda.anaconda.org/pytorch/osx-64/pytorch-1.4.0-py3.7_0.tar.bz2>
Elapsed: -

An HTTP error occurred when trying to retrieve this URL.
HTTP errors are often intermittent, and a simple retry will get you on your way.

CondaHTTPError: HTTP 000 CONNECTION FAILED for url <https://conda.anaconda.org/pytorch/osx-64/torchvision-0.5.0-py37_cpu.tar.bz2>
Elapsed: -

An HTTP error occurred when trying to retrieve this URL.
HTTP errors are often intermittent, and a simple retry will get you on your way.

複製代碼

這種狀況通常是 pytorch 和 torchvision 包下載失敗，咱們能夠選擇手動下載它們而後離線安裝。咱們從上面的提示中提取包下載 url連接進行下載，你能夠將下載url連接中的 https://conda.anaconda.org/pytorch/ 替換爲 https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/ 以提升下載速度。

以上圖中的版本爲例，下載成功後執行如下命令分別安裝 pytorch 和 torchvision 包：

# env_name 爲 Conda 環境名
   conda install --offline  env_name /path/to/pytorch-1.4.0-py3.7_0.tar.bz2
   conda install --offline  env_name /path/to/torchvision-0.5.0-py37_cpu.tar.bz2
複製代碼

這樣 Pytorch 的安裝就完成了，你能夠愉快地進行煉丹了。linux 和 windows 平臺相似。

5. 總結

今天介紹了若是利用 Conda 安裝 Pytorch ，解決 Pytorch 安裝失敗的一些問題。若是你在安裝中有什麼不明白能夠經過私信我。

關注公衆號：Felordcn獲取更多資訊

我的博客：https://felord.cn