windows10环境下安装pytorch

本文详细介绍了如何安装Anaconda,接着是Nvidia显卡驱动的安装,强调了CUDA和cuDNN版本需与显卡驱动匹配。在安装CUDA11.6后,由于兼容性问题选择了cuDNN相应版本。最后,针对CUDA11.7的环境,安装了Pytorch11.3版本进行验证,成功建立了深度学习环境。
摘要由CSDN通过智能技术生成

安装Anaconda

Anaconda的安装按照下载安装的方式即可。
anaconda下载地址
anaconda安装教程

Nvidia显卡驱动程序安装

使用驱动管理软件下载显卡驱动并安装或者到显卡官网下载。
查看显卡的CUDA版本,如下图所示:
显卡信息
可以看到显卡驱动的版本是517.00。
显卡的CUDA版本是11.7。

CUDA安装

下载地址
CUDA的安装版本要低于电脑显卡的版本。我的电脑的CUDA版本是11.7,安装11.7以下的CUDA版本即可。在这里选择的是11.6。

cuDNN安装

下载地址
选择对应的版本安装即可。cuDNN的版本要对应CUDA的版本。

Pytorch安装

下载地址
pytorch尽量安装比自己CUDA版本低的版本。在这里安装的是pytorch11.3版本。在这里安装11.6版本的pytorch时一直提示找不到包,所以安装了11.3。

验证环境安装

显示如图则成功。
true

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值