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


相关实践学习
部署Stable Diffusion玩转AI绘画(GPU云服务器)
本实验通过在ECS上从零开始部署Stable Diffusion来进行AI绘画创作,开启AIGC盲盒。
相关文章
|
4月前
|
PyTorch Linux 算法框架/工具
pytorch学习一:Anaconda下载、安装、配置环境变量。anaconda创建多版本python环境。安装 pytorch。
这篇文章是关于如何使用Anaconda进行Python环境管理,包括下载、安装、配置环境变量、创建多版本Python环境、安装PyTorch以及使用Jupyter Notebook的详细指南。
520 1
pytorch学习一:Anaconda下载、安装、配置环境变量。anaconda创建多版本python环境。安装 pytorch。
|
4月前
|
并行计算 PyTorch TensorFlow
Ubuntu安装笔记(一):安装显卡驱动、cuda/cudnn、Anaconda、Pytorch、Tensorflow、Opencv、Visdom、FFMPEG、卸载一些不必要的预装软件
这篇文章是关于如何在Ubuntu操作系统上安装显卡驱动、CUDA、CUDNN、Anaconda、PyTorch、TensorFlow、OpenCV、FFMPEG以及卸载不必要的预装软件的详细指南。
6524 3
|
4月前
|
PyTorch TensorFlow 算法框架/工具
Jetson环境安装(一):Ubuntu18.04安装pytorch、opencv、onnx、tensorflow、setuptools、pycuda....
本文提供了在Ubuntu 18.04操作系统的NVIDIA Jetson平台上安装深度学习和计算机视觉相关库的详细步骤,包括PyTorch、OpenCV、ONNX、TensorFlow等。
251 1
Jetson环境安装(一):Ubuntu18.04安装pytorch、opencv、onnx、tensorflow、setuptools、pycuda....
|
4月前
|
机器学习/深度学习 缓存 PyTorch
pytorch学习一(扩展篇):miniconda下载、安装、配置环境变量。miniconda创建多版本python环境。整理常用命令(亲测ok)
这篇文章是关于如何下载、安装和配置Miniconda,以及如何使用Miniconda创建和管理Python环境的详细指南。
856 0
pytorch学习一(扩展篇):miniconda下载、安装、配置环境变量。miniconda创建多版本python环境。整理常用命令(亲测ok)
|
4月前
|
并行计算 PyTorch 编译器
|
4月前
|
并行计算 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
184 0
|
4月前
|
并行计算 开发工具 异构计算
在Windows平台使用源码编译和安装PyTorch3D指定版本
【10月更文挑战第6天】在 Windows 平台上,编译和安装指定版本的 PyTorch3D 需要先安装 Python、Visual Studio Build Tools 和 CUDA(如有需要),然后通过 Git 获取源码。建议创建虚拟环境以隔离依赖,并使用 `pip` 安装所需库。最后,在源码目录下运行 `python setup.py install` 进行编译和安装。完成后即可在 Python 中导入 PyTorch3D 使用。
476 0
|
6月前
|
并行计算 TensorFlow 算法框架/工具
Window安装TensorFlow-GPU版本
Window安装TensorFlow-GPU版本
108 0
|
6月前
|
PyTorch 算法框架/工具 Python
安装anaconda配置pytorch虚拟环境遇到的问题及解决办法
本文介绍了在配置Anaconda时遇到`PackagesNotFoundError`的问题,并提供了通过添加`conda-forge`通道和创建指定Python版本的PyTorch虚拟环境来解决这个问题的方法。
|
3天前
|
存储 机器学习/深度学习 人工智能
2025年阿里云GPU服务器租用价格、选型策略与应用场景详解
随着AI与高性能计算需求的增长,阿里云提供了多种GPU实例,如NVIDIA V100、A10、T4等,适配不同场景。2025年重点实例中,V100实例GN6v单月3830元起,适合大规模训练;A10实例GN7i单月3213.99元起,适用于混合负载。计费模式有按量付费和包年包月,后者成本更低。针对AI训练、图形渲染及轻量级推理等场景,推荐不同配置以优化成本和性能。阿里云还提供抢占式实例、ESSD云盘等资源优化策略,支持eRDMA网络加速和倚天ARM架构,助力企业在2025年实现智能计算的效率与成本最优平衡。 (该简介为原文内容的高度概括,符合要求的字符限制。)

热门文章

最新文章