【学生向】PyTorch和CUDA有啥关系,怎么下载?

1、使用PyTorch进行深度学习时,为什么提到CUDA?什么是CUDA?它和PyTorch的关系是?

CUDA 是 NVIDIA 提供的用于 GPU 加速计算的平台和编程模型;PyTorch 是一个基于 Python 的科学计算库。可以将它们视为两个不同的软件,都需要单独安装,而它们的关系是:PyTorch 可以通过 CUDA 利用 GPU 加速计算。因此,如果你下载的PyTorch为CPU版本,进行深度学习就完全不需要下载CUDA;PyTorch为GPU版本时才需要再下载CUDA。我太懒了就直接选的PyTorch的CPU版本然后做实验(虽然我的笔记本也并没有独立GPU)。

2、PyTorch的下载

非常简单,进入PyTorch的官网,选择自己需要的选项,复制下面的语句粘贴到命令行运行即可。

3、CUDA的安装

进入官网,选择自己需要的选项,点击下载即可,然后安装下载好的.exe文件。

4、PyTorch的版本和CUDA的版本必须要匹配哦。

  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值