PyTorch原始碼安裝小記

2021-07-31 14:59:32 字數 1016 閱讀 9051

torch是facebook air去年6月開源的深度學習框架,向g家的tensorflow看起,基於lua開發,可以充分利用gpu進行計算加速。pytorch是fair今年1月發布的,在原來基礎上提供python介面支援?也是正面剛支援py的tf吧。

本來安裝pytorch比較簡單,按照官網,裝完anaconda後一句conda命令就可以安裝:

然而最後還是倒在牆下。

沒有辦法,只能選擇原始碼編譯安裝了:

對於ubuntu 14.04,把pytorch原始碼clone到本地後:

export cmake_prefix_path=[anaconda root directory]

# install basic dependencies

conda install numpy pyyaml mkl setuptools cmake gcc cffi

# add lapack support for the gpu

conda install -c soumith magma-cuda80 # or magma-cuda75 if cuda 7.5

python setup.py install

編譯完後急切想知道是否安裝成功,於是熟練地import torch,然後就會報錯:

importerror:no

module

named

_c

查了下pytorch repo的issues,果然很多人遇到:/issues/7

作者淡定說,不要在pytorch專案根目錄下匯入torch……

然後就可以了。

下一步就基於pytorch跑個小demo吧。

Windows原始碼安裝PyTorch 0 4

在windows安裝原始碼安裝pytorch 0.4。windows 10 home visual studio 2017 community cuda 9.0 cudnn 7.0 anaconda 5,python 3.6.4 需要安裝visual studio 2017,免費的community...

Windows原始碼安裝PyTorch 0 4

在windows安裝原始碼安裝pytorch 0.4。windows 10 home visual studio 2017 community cuda 9.0 cudnn 7.0 anaconda 5,python 3.6.4 需要安裝visual studio 2017,免費的community...

Pytorch原始碼注釋

field類為可以由張量表示的常見文字處理資料型別建模。它包含乙個vocab物件,用於定義字段元素的可能值集及其對應的數字表示。field物件還包含與資料型別應如何數位化有關的其他引數,例如標記化方法和應生成的tensor型別。如果在資料集中的兩列之間共享字段 例如,qa資料集中的問題和答案 則它們...