MacOSX安装Pytorch-GPU版本

简介: 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 图像分类演示的训练时间。


image.png

image.png

自己编译

如果您发现版本不能满足您的要求,您可以自己从源代码编译。


1.0.1.

1.0rc1.

源路径在source_pathesmaster 分支的文件夹中可用。

安装 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


相关实践学习
在云上部署ChatGLM2-6B大模型(GPU版)
ChatGLM2-6B是由智谱AI及清华KEG实验室于2023年6月发布的中英双语对话开源大模型。通过本实验,可以学习如何配置AIGC开发环境,如何部署ChatGLM2-6B大模型。
相关文章
|
11月前
|
PyTorch Linux 算法框架/工具
pytorch学习一:Anaconda下载、安装、配置环境变量。anaconda创建多版本python环境。安装 pytorch。
这篇文章是关于如何使用Anaconda进行Python环境管理,包括下载、安装、配置环境变量、创建多版本Python环境、安装PyTorch以及使用Jupyter Notebook的详细指南。
1294 1
pytorch学习一:Anaconda下载、安装、配置环境变量。anaconda创建多版本python环境。安装 pytorch。
|
11月前
|
并行计算 PyTorch TensorFlow
Ubuntu安装笔记(一):安装显卡驱动、cuda/cudnn、Anaconda、Pytorch、Tensorflow、Opencv、Visdom、FFMPEG、卸载一些不必要的预装软件
这篇文章是关于如何在Ubuntu操作系统上安装显卡驱动、CUDA、CUDNN、Anaconda、PyTorch、TensorFlow、OpenCV、FFMPEG以及卸载不必要的预装软件的详细指南。
10864 4
|
6月前
|
机器学习/深度学习 并行计算 PyTorch
Windows下CUDA+pytorch安装
以下是关于在Windows下安装CUDA和PyTorch的简要介绍及参考链接:
328 0
Windows下CUDA+pytorch安装
|
11月前
|
PyTorch TensorFlow 算法框架/工具
Jetson环境安装(一):Ubuntu18.04安装pytorch、opencv、onnx、tensorflow、setuptools、pycuda....
本文提供了在Ubuntu 18.04操作系统的NVIDIA Jetson平台上安装深度学习和计算机视觉相关库的详细步骤,包括PyTorch、OpenCV、ONNX、TensorFlow等。
934 1
Jetson环境安装(一):Ubuntu18.04安装pytorch、opencv、onnx、tensorflow、setuptools、pycuda....
|
11月前
|
机器学习/深度学习 缓存 PyTorch
pytorch学习一(扩展篇):miniconda下载、安装、配置环境变量。miniconda创建多版本python环境。整理常用命令(亲测ok)
这篇文章是关于如何下载、安装和配置Miniconda,以及如何使用Miniconda创建和管理Python环境的详细指南。
3525 0
pytorch学习一(扩展篇):miniconda下载、安装、配置环境变量。miniconda创建多版本python环境。整理常用命令(亲测ok)
|
11月前
|
并行计算 PyTorch 编译器
|
11月前
|
并行计算 Ubuntu 算法
Ubuntu18 服务器 更新升级CUDA版本 pyenv nvidia ubuntu1804 原11.2升级到PyTorch要求12.1 全过程详细记录 apt update
Ubuntu18 服务器 更新升级CUDA版本 pyenv nvidia ubuntu1804 原11.2升级到PyTorch要求12.1 全过程详细记录 apt update
347 0
|
11月前
|
并行计算 开发工具 异构计算
在Windows平台使用源码编译和安装PyTorch3D指定版本
【10月更文挑战第6天】在 Windows 平台上,编译和安装指定版本的 PyTorch3D 需要先安装 Python、Visual Studio Build Tools 和 CUDA(如有需要),然后通过 Git 获取源码。建议创建虚拟环境以隔离依赖,并使用 `pip` 安装所需库。最后,在源码目录下运行 `python setup.py install` 进行编译和安装。完成后即可在 Python 中导入 PyTorch3D 使用。
1069 0
|
并行计算 TensorFlow 算法框架/工具
Window安装TensorFlow-GPU版本
Window安装TensorFlow-GPU版本
296 0
|
4月前
|
机器学习/深度学习 PyTorch API
PyTorch量化感知训练技术:模型压缩与高精度边缘部署实践
本文深入探讨神经网络模型量化技术,重点讲解训练后量化(PTQ)与量化感知训练(QAT)两种主流方法。PTQ通过校准数据集确定量化参数,快速实现模型压缩,但精度损失较大;QAT在训练中引入伪量化操作,使模型适应低精度环境,显著提升量化后性能。文章结合PyTorch实现细节,介绍Eager模式、FX图模式及PyTorch 2导出量化等工具,并分享大语言模型Int4/Int8混合精度实践。最后总结量化最佳策略,包括逐通道量化、混合精度设置及目标硬件适配,助力高效部署深度学习模型。
586 21
PyTorch量化感知训练技术:模型压缩与高精度边缘部署实践

热门文章

最新文章

推荐镜像

更多