Pytorch入門 安裝

2021-08-08 11:24:43 字數 937 閱讀 5741

pytorch目前支援的平台有linux和osx,在pytorch官網上每種平台提供了conda、pip、source三種安裝方式,同時也可以根據有無gpu進行cuda安裝,在這裡以ubuntu14.04進行安裝學習。

1. anaconda安裝配置 

安裝過程參考我之前的anaconda+tensorflow+theano+keras安裝部落格。 

由於牆的問題,用conda安裝pytorch過程中會連線失敗,這是因為anaconda.org的伺服器在國外。在這裡可以用清華tuna映象源,包含anaconda倉庫的映象,將其加入conda的配置,配置如下:

# 新增anaconda的tuna映象

$ conda config --add channels

.tsinghua

.edu

.cn/anaconda/pkgs/free/

# tuna的help中映象位址加有引號,需要去掉

# 設定搜尋時顯示通道位址

$ conda config --set show_channel_urls yes

執行完上述命令後,會生成~/.condarc檔案,記錄著對conda的配置,直接手動建立、編輯該檔案是相同的效果。

2. pytorch安裝

在這裡的安裝,我採用conda安裝: 

$  conda install pytorch torchvision -c soumith
3. 測試

進入python模式下,看能否匯入torch成功:

$  python

> import torch

PyTorch學習 安裝PyTorch

例如,使用的是 windows 系統,想用 pip 安裝,python 是 3.6 版的,沒有 gpu 加速,那就按上面的選,然後根據上面的提示,在 terminal 中輸入以下指令就好了 pip3 install torch 1.3.1 cpu torchvision 0.4.2 cpu ftor...

PyTorch快速入門

詳細的pytorch教程可以去pytorch官網的學習指南進一步學習,下面主要對pytorch做簡單的介紹,能夠快速入門。首先pytorch是基於python的科學計算類庫,主要有以下兩個方面的應用 作為numpy的替代者,充分利用gpu的計算能力。提供乙個靈活 快速的深度學習平台。tensor 與...

Pytorch快速入門

附錄qi 學習經歷 博主在半年內近乎從零開始了解深度學習,並直接跳過了機器學習的內容,沒有系統的上過與深度學習有關的課程,也是第一次上手pythorch,且以前也很少用python進行程式設計.幾乎都在用matlab與c 因此一開始接觸起來很吃力,甚至走了不少彎路,在這裡將自己通過學習與交流得到的一...