win10 pytorch安装(cuda + cudnn、anaconda、pytorch )


网上的资料太乱,自己摸索了好几次,算是能够顺利完成pytorch的环境配置,把自己的过程整理一下,一方面是加强理解记忆,一方面希望能够对需要的人有所帮助

一、CUDA + cuDNN

CUDA
CUDA是NVIDIA推出的用于自家GPU的并行计算框架,也就是说CUDA只能在NVIDIA的GPU上运行,而且只有当要解决的计算问题是可以大量并行计算的时候才能发挥CUDA的作用。

cuDNN
cuDNN(CUDA Deep Neural Network library):是NVIDIA打造的针对深度神经网络的加速库,是一个用于深层神经网络的GPU加速库。如果你要用GPU训练模型,cuDNN不是必须的,但是一般会采用这个加速库。

CUDA和cuDNN直接在官网下载即可,目前网速还不错:
CUDA的下载链接:https://developer.nvidia.com/cuda-toolkit-archive
cuDNN的下载链接:https://developer.nvidia.com/rdp/cudnn-download

1、CUDA

官方文档里给出了CUDA版本和电脑显卡驱动版本之间的兼容关系:
https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html
在这里插入图片描述因此,CUDA 版本的选择要根据电脑的驱动版本来定
在这里插入图片描述
在这里插入图片描述
在这里插入图片描述
CUDA的下载链接:https://developer.nvidia.com/cuda-toolkit-archive
根据显卡驱动版本就可以去下载了

2、cuDNN

cuDNN的下载十分简单,直接根据安装的CUDA版本进行下载就行
在这里插入图片描述下载解压后如下图
在这里插入图片描述
直接将这些文件夹的内容拷贝到,相应的CUDA安装目录下对应的文件夹就行
一般安装目录在C盘,比如我的是:
在这里插入图片描述

二、anaconda(未完待续)

三、pytorch(未完待续)

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值