2019 Mac13.6安装cuda+cudnn+pytorch

我的原文:https://www.hijerry.cn/p/49582.html

本次主要记录一些自己遇到的问题的解决方案

CUDA、cuDNN

大致流程参照 2018 MAC安装CUDA、cuDNN(Gaming Box1070)

顺序是:GPU Driver、CUDA Driver、CUDA Toolkit、cuDNN

安装驱动时要注意:

MacOS与NVIDIA GPU Driver的版本要匹配,才能驱动显卡
CUDA Driver与NVIDIA GPU Driver的版本要一致,CUDA才能找到显卡

关键地址:

### 如何在PyTorch中配置和使用CUDACuDNN #### 安装合适的CUDACuDNN版本 为了使PyTorch能够利用GPU加速计算,需要确保安装了兼容的CUDACuDNN版本。表展示了不同版本之间的对应关系[^2]。 对于最新的PyTorch版本(例如1.10.x),推荐使用的CUDA版本为11.3或更高;而对于较旧版本,则应选择相应的较低版次的CUDA工具包来匹配特定需求的应用场景。 #### 创建并激活Conda虚拟环境 通过Anaconda创建一个新的Python环境可以简化依赖管理过程: ```bash conda create -n pytorch_env python=3.8 conda activate pytorch_env ``` 这一步骤有助于隔离项目所需的库和其他全局安装软件之间可能存在的冲突问题[^1]。 #### 安装PyTorch及相关组件 一旦选择了正确的CUDA/CuDNN组合之后,在上述环境中可以通过pip命令轻松获取预编译好的二进制发行版: ```bash pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu113 ``` 这里假设已经决定采用CUDA 11.3作为目标平台,并指定了`cu113`参数以获得相应支持。如果选用其他版本,请调整URL中的编号部分以反映实际需求。 #### 验证安装是否成功 完成以上操作后,可通过简单的测试脚本来确认一切正常工作: ```python import torch print(f"CUDA Available: {torch.cuda.is_available()}") if torch.cuda.is_available(): device = 'cuda' else: device = 'cpu' tensor_example = torch.tensor([1., 2., 3.], device=device) print(tensor_example) ``` 这段代码会打印出当前系统上是否有可用的CUDA设备以及尝试在一个张量对象上调用它来进行基本运算。如果有任何错误发生或者输出显示无法访问GPU资源,则表明可能存在未解决的问题需要进一步排查。
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值