Pytorch學習筆記

簡介

• pytorch
• pytorch github

pytorch 包含了很多子庫, 包括：pytorch (torch), vision(torchvision), audio, text , 具體參見 https://github.com/pytorch .

安裝

• 源碼:
  • pytorch
  • torchvision
• 安裝包:
  • pytorch
  • torchvision
• 文檔:

安裝包方式

在線安裝

參考 START LOCALLY <https://pytorch.org/get-started/locally/>_ , 選擇環境, 執行安裝命令即可.

conda install pytorch torchvision cudatoolkit=9.0 -c pytorch

本地安裝包安裝

1. Anaconda安裝

conda install --use-local ./pytorch-1.1.0-py3.7_cuda9.0.176_cudnn7.5.1_0.tar.bz2 -c pytorch
conda install torchvision cudatoolkit=9.0 -c pytorch

2. Python安裝

從 pypi <https://pypi.org/>_ 下載對應版本的安裝包, 執行 pip install 命令安裝即可, 如

    sudo pip3 install torch-1.1.0-cp35-cp35m-manylinux1_x86_64.whl
    sudo pip3 install torchvision

源碼方式

• 官方參考： pytorch from source
• 官方參考： torch vision from source

參照官方說明安裝即可。

1. pytorch

conda install numpy pyyaml mkl mkl-include setuptools cmake cffi typing

# Add LAPACK support for the GPU if needed
conda install -c pytorch magma-cuda90 # or [magma-cuda80 | magma-cuda91] depending on your cuda version

git clone --recursive https://github.com/pytorch/pytorch
cd pytorch

export CMAKE_PREFIX_PATH=${CONDA_PREFIX:-"$(dirname $(which conda))/../"}
python setup.py install

2. torchvision

python setup.py install

這是各大DL平臺中安裝最爲簡單和順利的安裝體驗了.

構建文檔

1. pytorch

Sphinx格式文檔

cd docs/
pip install -r requirements.txt

make html
make epub

2. torchvision

cd docs/
pip install -r requirements.txt

make html
make epub

使用

問題與解決

安裝

PyQt

ImportError: anaconda3/lib/python3.6/site-packages/PyQt5/../../../libQt5Core.so.5: version `Qt_5.9' not found (required by /home/liu/anaconda3/lib/python3.6/site-packages/PyQt5/QtCore.so)

matplotlib

pip install PyQt

使用

找不到 torch

參照 這裏，即安裝沒有問題， 在另一個不含torch的目錄使用即可。

python
import torch
# 提示如下錯誤信息
ModuleNotFoundError: No module named 'torch._C'

CUDA相關

無論是通過Anaconda還是pip安裝完畢PyTorch後, 在Python解釋器中導入torch( import torch )均會報出如下錯誤

    from torch._C import * ImportError: libcurand.so.8.0: cannot open shared object file

系統爲 Ubuntu16.04LTS, Python3.5, Python3.7(Anaconda), CUDA9.0 和 CUDA8.0 共存. 安裝前環境已經切換爲 CUDA9.0, 上述錯誤提示找不到 CUDA8.0 相關文件, 因而懷疑一些編譯的庫使用的是CUDA8.0. 最終發現是之前安裝的 caffe2 的影響, 導致 PyTorch 中的cafe2不能正確安裝, 卸載之前安裝的 caffe2, 問題解決.

找不到 torchvision

可能沒安裝或者安裝失敗，請注意環境，如果是 conda環境，要在相應環境下執行，如果不是，要退出conda環境。

>>> import torch
>>> import torchvision
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
ModuleNotFoundError: No module named 'torchvision'