pytorch安装: WIN10 + cuDA 10.1
目录
一,安装cuDA,cuDNN
1,下载cunDA
cuDA下载地址

2,安装cuDA
cuDA下载完成以后,双击启动安装过程。具体流程与其他软件相似,就是时间有点长。另外,安装目录按照默认就行。
3,下载cuDNN
cuDNN下载地址
要下载cuDNN,首先要注册账号(我的注册名为jeffrey),然后点击首页的>PLATFORMS:

接下来就可以下载了:
解压下载的cuDNN压缩包,将其中的三个文件夹复制到cuDA的安装目录中,替换掉cuDA中的同名文件夹。


4,添加环境变量;
步骤:我的电脑 -> 系统属性 -> 系统信息 -> 高级系统设置 -> 环境变量 -> 系统变量中的path 点击进行 编辑 -> 添加如下图所示的路径(具体情况具体分析)



本文提供了一步一步的PyTorch安装指南,包括cuDA和cuDNN的安装,详细讲解了环境变量配置,以及torch和torchvision的下载与安装,确保100%成功率。
最低0.47元/天 解锁文章
852

被折叠的 条评论
为什么被折叠?



