MacOSX安装Pytorch-GPU版本
Pytorch OSX 构建
不幸的是，Pytorch 团队没有发布支持 CUDA 的 Mac OS 二进制包。该项目提供现成的二进制包。Both Python 2.7 and 3.7 are supported now!
很不幸，Pytorch 团队不发布 Mac OS CUDA 版本。本项目提供 Mac OS 上编译好、可直接安装的 Pytorch CUDA 版本。本项目同时支持Python 2.7 和 3.7 了！
性能警告
【2020.02.18】我在 Mac OS X 10.13.6 和 Ubuntu 16.04 上用 CUDA 10.1 和 CUDNN 7.6.5 对 Pytorch 1.3.1 进行了基准测试，在 Mac OS 上的性能大约是在 Ubuntu 上的 2/3。此外，在 Mac OS 上更容易遇到“CUDA OUT OF MEMORY”错误，因为操作系统需要大量的 GPU 内存来显示。请注意这种性能差异，如果您要处理大量数据，最好转向 Ubuntu！
下表列出了 MNIST 图像分类演示的训练时间。
自己编译
如果您发现版本不能满足您的要求，您可以自己从源代码编译。
安装 Python 2.7
首先，确保您的 CUDA 驱动程序和 cudnn 安装正确，并将文件夹中的依赖项复制usr_local_lib到 path /usr/local/lib。另外，使用 Homebrew 安装 OpenMP。
首先，确保 CUDA 驱动和cudnn正确安装，并且将其中usr_local_lib的依赖项复制到路径/usr/local/lib。也要通过 Homebrew 安装 OpenMP。
sudo mkdir /usr/local
sudo mkdir /usr/local/lib
sudo cp usr_local_lib/* /usr/local/lib/
brew install libomp
brew link --overwrite libomp
二、卸载之前的pytorch安装
即将，卸载之前版本的pytorch：
pip uninstall torch
从这个项目安装wheel包：
安装：
pip install torch*.whl
安装 torchvision：
pip install -U torchvision
安装 Python 3
安装从自制的Python 3.x的第一，然后只要按照用于Python 2.7导向和替换pip用命令pip3和python用python3。
首先从Homebrew安装Python 3.x，然后按照Python 2.7的安装步骤执行，注意将pip替换为pip3，并用python3启动python。
享受！
开始使用Pytorch吧！
源代码
源代码来自：https://github.com/pytorch/pytorch
相关链接
如果您需要 osx 的 Tensorflow 构建，请转到此页面：
https://github.com/TomHeaven/tensorflow-osx-build
如果您需要为 osx 构建 MxNet，请转到此页面：
https://github.com/TomHeaven/mxnet_osx_build
如果你需要 Tensorflow 包，请看这个页面：https://github.com/TomHeaven/tensorflow-osx-build
如果你需要 MxNet 包，请看这个页面：https://github.com/TomHeaven/mxnet_osx_build
「
我们坐在灯上
我们火光通明
我们做梦的胳膊搂在一起
我们栖息的桌子飘向麦地
我们安坐的灯火涌向星辰
」
文章来源: blog.csdn.net，作者：府学路18号车神，版权归原作者所有，如需转载，请联系作者。
原文链接：blog.csdn.net/weixin_44333889/article/details/118694271
- 点赞
- 收藏
- 关注作者
评论（0）