GPU版本的PyTorch安装与环境配置

前言:笔者最近打算安装pytorch学习一下,由于各种原因花了一天时间捣腾,最终发现亲测有效的一种方法,安装pytorch需要注意各个部分版本之间的兼容性。GPU版本的PyTorch安装与环境配置一、安装需求流程二、 步骤1: pytorch与cudatoolkit驱动器的安装与下载2.1 cudatoolkit驱动器下载2.2 torch下载三、步骤2: CUDA与CUDNN的安装与环境配置3.1 CUDA下载安装和配置3.2 安装CUDNN四、验证torch安装和是否能使用GPU一、安装需求流程
摘要由CSDN通过智能技术生成

前言:笔者最近打算安装pytorch学习一下,由于各种原因花了一天时间捣腾,最终发现亲测有效的一种方法,安装pytorch需要注意各个部分版本之间的兼容性。

一、安装需求流程

GPU版本pytorch安装需求:

  1. cudatoolkit驱动器的安装与下载
  2. pytorch的安装与下载
  3. cuda的下载与环境配置
  4. cudnn的下载与环境配置

因为我们是下载GPU版本的pytorch,因此需要注意你的显卡的配置是否和cuda驱动匹配,此外cudatoolkit与cudnn也需要和cudatoolkit驱动版本相兼容.

二、 步骤1: pytorch与cudatoolkit驱动器的安装与下载

2.1 cudatoolkit驱动器下载

在cmd中使用nvidia-smi查询可以安装的驱动版本在这里插入图片描述
然后用通过表1和表2确定CUDA驱动和pytorch的版本,要求Driver Versiond>384.81的cudatoolkit,也需要注意后面的cuda安装也要和cuda驱动版本兼容.由于我个人配置的电脑Driver Versiond=430.(非截图)因此我选择下载cuda驱动的版本为10.2.
在这里插入图片描述在这里插入图片描述
疑难问题:pytorch与cudatoolkit的下载
一般通过pytorch官网给出的命令下载, pytorch官网. 通过官网给出的命令在anaconda的conda中输入该命令下载
问题1:墙内,网络下载慢,即使加载清华源,也需要尝试下载多次才可以成功.
问题2:我使用清华源下载后的pytorch版本为CPU版本,不是GPU版本.在这里插入图片描述
我先给出一般cuda下载方式&

  • 43
    点赞
  • 184
    收藏
    觉得还不错? 一键收藏
  • 4
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值