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 算法框架/工具 Python
pytorch安装教程
pytorch安装教程
122 0
|
3月前
|
PyTorch 算法框架/工具
win10下安装pytorch,torchvision遇到的bug
win10下安装pytorch,torchvision遇到的bug
30 1
|
1月前
|
机器学习/深度学习 人工智能 PyTorch
【Deepin 20深度探索】一键解锁Linux深度学习潜能:从零开始安装Pytorch,驾驭AI未来从Deepin出发!
【8月更文挑战第2天】随着人工智能的迅猛发展,深度学习框架Pytorch已成为科研与工业界的必备工具。Deepin 20作为优秀的国产Linux发行版,凭借其流畅的用户体验和丰富的软件生态,为深度学习爱好者提供理想开发平台。本文引导您在Deepin 20上安装Pytorch,享受Linux下的深度学习之旅。
66 12
|
1月前
|
并行计算 TensorFlow 算法框架/工具
Windows11+CUDA12.0+RTX4090如何配置安装Tensorflow2-GPU环境?
本文介绍了如何在Windows 11操作系统上,配合CUDA 12.0和RTX4090显卡,通过创建conda环境、安装特定版本的CUDA、cuDNN和TensorFlow 2.10来配置TensorFlow GPU环境,并提供了解决可能遇到的cudnn库文件找不到错误的具体步骤。
186 3
|
1月前
|
并行计算 TensorFlow 算法框架/工具
Window安装TensorFlow-GPU版本
Window安装TensorFlow-GPU版本
37 0
|
1月前
|
PyTorch 算法框架/工具 Python
安装anaconda配置pytorch虚拟环境遇到的问题及解决办法
本文介绍了在配置Anaconda时遇到`PackagesNotFoundError`的问题,并提供了通过添加`conda-forge`通道和创建指定Python版本的PyTorch虚拟环境来解决这个问题的方法。
|
1月前
|
并行计算 Ubuntu PyTorch
【Deepin 20系统】Linux系统Anaconda安装Pytorch
本文提供了在Deepin 20系统上使用Anaconda安装Pytorch的详细步骤,包括创建一个新的conda环境、检查系统Cuda版本、选择对应Cuda版本的Pytorch安装命令、添加镜像源以加速下载,以及执行安装命令。
44 1
|
1月前
|
PyTorch 算法框架/工具
Pytorch 安装
Pytorch 安装
28 0
|
2月前
|
Linux TensorFlow 算法框架/工具
安装GPU版本的TensorFlow
【7月更文挑战第3天】安装GPU版本的TensorFlow。
127 1
|
3月前
|
机器学习/深度学习 并行计算 PyTorch
安装PyTorch详细步骤
安装PyTorch时,选择CPU或GPU版本。有Nvidia显卡需装CUDA和cuDNN,可从NVIDIA官网下载CUDA 11.8和对应版本cuDNN。无Nvidia显卡则安装CPU版。安装PyTorch通过conda或pip,GPU版指定`cu118`或`rocm5.4.2`镜像源。验证安装成功使用`torch._version_`和`torch.cuda.is_available()`。

热门文章

最新文章