Pytorch学习笔记

简介

• pytorch
• pytorch github

pytorch 包含了很多子库, 包括：pytorch (torch), vision(torchvision), audio, text , 具体参见 https://github.com/pytorch .

安装

• 源码:
  • pytorch
  • torchvision
• 安装包:
  • pytorch
  • torchvision
• 文档:

安装包方式

在线安装

参考 START 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 _ 下载对应版本的安装包, 执行 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 "", line 1, in <module>
ModuleNotFoundError: No module named 'torchvision'