Pytorch学习笔记

简介

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

安装

安装包方式

在线安装

参考 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
  1. Python安装

pypi <https://pypi.org/>_ 下载对应版本的安装包, 执行 pip install 命令安装即可, 如

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

源码方式

参照官方说明安装即可。

  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
  1. torchvision
python setup.py install

这是各大DL平台中安装最为简单和顺利的安装体验了.

构建文档

  1. pytorch

Sphinx格式文档

cd docs/
pip install -r requirements.txt

make html
make epub
  1. 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'
发布了84 篇原创文章 · 获赞 131 · 访问量 41万+
展开阅读全文

没有更多推荐了,返回首页

©️2019 CSDN 皮肤主题: 大白 设计师: CSDN官方博客

分享到微信朋友圈

×

扫一扫,手机浏览