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

本文详细介绍了如何在GPU环境下安装PyTorch,包括cudatoolkit驱动器、CUDA、CUDNN的下载与配置,以及验证PyTorch是否能使用GPU。在安装过程中,需要注意各组件版本之间的兼容性,例如cuda驱动版本需大于384.81,并与cudatoolkit、cudnn相匹配。文章提供了具体的下载地址、安装步骤及环境变量配置方法。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

前言:笔者最近打算安装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下载方式,之后再给出单独下载GPU版本pytorch的方式

评论 4
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值