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大模型。
相关文章
|
12月前
|
PyTorch Linux 算法框架/工具
pytorch学习一:Anaconda下载、安装、配置环境变量。anaconda创建多版本python环境。安装 pytorch。
这篇文章是关于如何使用Anaconda进行Python环境管理,包括下载、安装、配置环境变量、创建多版本Python环境、安装PyTorch以及使用Jupyter Notebook的详细指南。
1386 1
pytorch学习一:Anaconda下载、安装、配置环境变量。anaconda创建多版本python环境。安装 pytorch。
|
12月前
|
并行计算 PyTorch TensorFlow
Ubuntu安装笔记(一):安装显卡驱动、cuda/cudnn、Anaconda、Pytorch、Tensorflow、Opencv、Visdom、FFMPEG、卸载一些不必要的预装软件
这篇文章是关于如何在Ubuntu操作系统上安装显卡驱动、CUDA、CUDNN、Anaconda、PyTorch、TensorFlow、OpenCV、FFMPEG以及卸载不必要的预装软件的详细指南。
10982 4
|
7月前
|
机器学习/深度学习 并行计算 PyTorch
Windows下CUDA+pytorch安装
以下是关于在Windows下安装CUDA和PyTorch的简要介绍及参考链接:
360 0
Windows下CUDA+pytorch安装
|
12月前
|
PyTorch TensorFlow 算法框架/工具
Jetson环境安装(一):Ubuntu18.04安装pytorch、opencv、onnx、tensorflow、setuptools、pycuda....
本文提供了在Ubuntu 18.04操作系统的NVIDIA Jetson平台上安装深度学习和计算机视觉相关库的详细步骤,包括PyTorch、OpenCV、ONNX、TensorFlow等。
978 1
Jetson环境安装(一):Ubuntu18.04安装pytorch、opencv、onnx、tensorflow、setuptools、pycuda....
|
12月前
|
机器学习/深度学习 缓存 PyTorch
pytorch学习一(扩展篇):miniconda下载、安装、配置环境变量。miniconda创建多版本python环境。整理常用命令(亲测ok)
这篇文章是关于如何下载、安装和配置Miniconda,以及如何使用Miniconda创建和管理Python环境的详细指南。
3783 0
pytorch学习一(扩展篇):miniconda下载、安装、配置环境变量。miniconda创建多版本python环境。整理常用命令(亲测ok)
|
12月前
|
并行计算 PyTorch 编译器
|
12月前
|
并行计算 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
360 0
|
12月前
|
并行计算 开发工具 异构计算
在Windows平台使用源码编译和安装PyTorch3D指定版本
【10月更文挑战第6天】在 Windows 平台上,编译和安装指定版本的 PyTorch3D 需要先安装 Python、Visual Studio Build Tools 和 CUDA(如有需要),然后通过 Git 获取源码。建议创建虚拟环境以隔离依赖,并使用 `pip` 安装所需库。最后,在源码目录下运行 `python setup.py install` 进行编译和安装。完成后即可在 Python 中导入 PyTorch3D 使用。
1117 0
|
并行计算 TensorFlow 算法框架/工具
Window安装TensorFlow-GPU版本
Window安装TensorFlow-GPU版本
315 0
|
16天前
|
机器学习/深度学习 数据采集 人工智能
PyTorch学习实战:AI从数学基础到模型优化全流程精解
本文系统讲解人工智能、机器学习与深度学习的层级关系,涵盖PyTorch环境配置、张量操作、数据预处理、神经网络基础及模型训练全流程,结合数学原理与代码实践,深入浅出地介绍激活函数、反向传播等核心概念,助力快速入门深度学习。
69 1

热门文章

最新文章

推荐镜像

更多