【Pytorch-gpu版安装教程,针对cuda11.1】

针对cuda11.1版本安装pytorch-gpu版本@TOC

配了3天的环境终于true了,简单列举一下这三天遇到的问题和解决方法。
首先介绍一下成功的路径。
由于校园网不好,我使用了离线下载。
在下载torch之前,最好先打开命令行(输入cmd),在命令行中输入nvidia-smi查看显卡对应的版本:
在这里插入图片描述
然后打开这个网址https://download.pytorch.org/whl/torch_stable.html,寻找版本对应的torch和torchvision进行下载。
此处介绍gpu版本,因为有英伟达显卡,所以选择下载gpu版本,此处我下载的是这两个
cu111/torch-1.9.0%2Bcu111-cp38-cp38-win_amd64.whl
cu111/torchvision-0.10.0%2Bcu111-cp38-cp38-win_amd64.whl
(cu111是对应cuda11.1),选择windows版本而不是linux,因为这里我有遇到过,先下载了linux版本然后报错,因为文件里只能识别win_amd64类似的名字。所以要选对版本。
我是将它们下载到了D盘的安装包里,然后在anaconda prompt中首先要进入存放这两个包的地址中,然后新建虚拟地址

  • 0
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 2
    评论
### 回答1: PyTorch版本CUDA版本之间的对应关系并不固定,需要查看PyTorch的官方文档。 根据PyTorch的官方文档,在2021年的最新版本PyTorch 1.7.0中,支持CUDA 11.0和11.1。因此,可以使用PyTorch 1.7.0与CUDA 11.1配合使用。 不过,建议您确认您的计算机环境是否符合PyTorch 1.7.0的系统要求,以确保安装和使用的顺利。 ### 回答2: 在CUDA 11.1的情况下,可以使用PyTorch 1.7.0或更高版本进行GPU加速的深度学习任务。PyTorch是一个基于Python的开源机器学习框架,它提供了丰富的工具和库,使得在GPU上进行高效的计算成为可能。 CUDA 11.1是NVIDIA的一个GPU加速计算平台,它提供了用于并行计算的许多功能和工具。PyTorch利用CUDA架构的强大性能,通过与CUDA 11.1的集成,能够在GPU上进行高效的深度学习计算。 PyTorch 1.7.0是支持CUDA 11.1版本,它支持使用CUDA 11.1进行加速的功能。这个版本是经过pytorch官方团队和社区的努力开发和测试的。使用PyTorch 1.7.0可以利用CUDA 11.1提供的计算能力,使得深度学习任务在GPU上运行得更快。 总而言之,CUDA 11.1PyTorch 1.7.0是相互兼容的。通过使用PyTorch 1.7.0,可以使用CUDA 11.1提供的强大的GPU加速功能来加速深度学习任务的计算。 ### 回答3: CUDA 11.1 对应PyTorch 版本是 1.7.0。具体来说,这个版本PyTorch是在2020年10月29日发布的,专门为CUDA 11.1进行了优化。此版本PyTorch支持CUDA 11.1的全部功能和特性,可以充分利用CUDA 11.1的性能提升。 PyTorch是一个开源的深度学习框架,提供了灵活的工具和库,用于构建和训练深度神经网络模型。它基于Python语言,利用CUDA加速计算,能够利用GPU的强大计算能力来加速训练过程。 CUDA是NVIDIA提供的一个并行计算平台和编程模型,它可以通过显卡的并行计算单元来加速计算。CUDA 11.1CUDA平台的一个版本,提供了一些新的功能和性能优化,可以进一步提升GPU计算的速度和效率。 PyTorch1.7.0是针对CUDA 11.1进行了优化的版本,可以充分利用CUDA 11.1的性能提升。用户在使用PyTorch构建深度学习模型时,可以选择使用CUDA 11.1PyTorch 1.7.0的组合,以充分发挥GPU的计算能力,加快训练速度和提高模型性能。

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值