git clone --recursive
pip install -r requirements.txt (這步操作超級慢)
python setup.py install
安裝成功後,測試:
python
>>>import torch
出現錯誤:
參考:cd /usr/local/lib/python3.5/dist-packages/torch/
ls發現 _c.cpython-36m-x86_64-linux-gnu.so, _dl.cpython-36m-x86_64-linux-gnu.so兩個so檔案的檔名都太長了,需要進行修改。
採用cp 的方式複製兩個檔案:
/usr/local/lib/python3.5/dist-packages/torch/$ sudo cp _c.cpython-36m-x86_64-linux-gnu.so _c.so
/usr/local/lib/python3.5/dist-packages/torch/$ sudo cp _dl.cpython-36m-x86_64-linux-gnu.so _dl.so
成功解決
pytorch 學習遇到的 問題
relu 和 leaky relu 中 inplace 設定 nn.relu inplace true default inplace false nn.leakyrelu inplace true default inplace false 將計算得到的值直接覆蓋之前的值 有時能夠節省運算記憶體,...
pytorch 遇到的小問題總結
一 transpose 是permute的精簡版本。transpose dim1,dim2 只能將兩個維度進行互換 y y.permute 2,3,1,4 按照2,3,1,4維進行重排列二 對兩個variable進行concat操作,按道理實現方式是c torch.cat a,b dim 0 但提示...
pytorch原始碼編譯遇到的問題
1.編譯到大概33 處,出現failed to run bash tools build pytorch libs.sh use cuda use nnpack nccl caffe2 libshm gloo c10d thd 解決 最新的pytorch原始碼編譯要求cudnn 5,公升級cudnn...