The value of knowledge lies not in possession, but in share.

0%

在Mac中搭建PyTorch网络框架

近期一直在研究PyTorch这个网络,发现真的是很好用啊!

首先,网络比较轻量;此外,网络框架的灵活性也比较高。相比于TensorFlow真的是很好入手,而且逻辑结构十分清晰!为Facebook点个赞!

在macOS Majave 10.14.4版本下进行PyTorch的安装,过程中遇到一些问题,在此对安装的过程及解决问题的方法进行一个记录。

Pip安装PyTorch

根据PyTorch官网上的说明

指令为:pip3 install torch torchvision

安装完毕后对PyTorch进行测试,报错如下:

1
2
3
4
5
6
7
8
>>> import torch
Traceback (most recent call last):
File "<stdin>", line 1, in <module>
File "/Library/Frameworks/Python.framework/Versions/3.6/lib/python3.6/site-packages/torch/__init__.py", line 79, in <module>
from torch._C import *
ImportError: dlopen(/Library/Frameworks/Python.framework/Versions/3.6/lib/python3.6/site-packages/torch/_C.cpython-36m-darwin.so, 9): Library not loaded: /usr/local/opt/libomp/lib/libomp.dylib
Referenced from: /Library/Frameworks/Python.framework/Versions/3.6/lib/python3.6/site-packages/torch/lib/libshm.dylib
Reason: image not found

提示没有libomp.dylib这个库。

1
brew install libomp

然后进一步测试:

1
2
3
4
5
Python 3.6.5 (v3.6.5:f59c0932b4, Mar 28 2018, 05:52:31) 
[GCC 4.2.1 Compatible Apple LLVM 6.0 (clang-600.0.57)] on darwin
Type "help", "copyright", "credits" or "license" for more information.
>>> import torch
>>>

问题解决!

安装HomeBrew (可选)

若Mac中没有安装brew工具,请参考此处的教程。

Homebrew是基于Ruby的(Mac电脑默认是安装的1.8.7版本的ruby),直接将下面的代码粘贴到Terminal中执行:

1
/usr/bin/ruby -e "$(curl -fsSL https://raw.githubusercontent.com/Homebrew/install/master/install)"

过程中需要多次回车进行确认。

🍭支持一根棒棒糖吧!