如何在anaconda里面配置pytorch环境(Win10+驱动程序版本11.6+ 下载的是pytorch1.7.0+cudatoolkit11.0+torchvision 0.8.0)

本文详细介绍了如何在Windows 10系统中安装PyTorch,包括检查CUDA驱动版本,选择兼容的CUDAToolkit 11.0.3,下载并安装CUDA,接着安装cuDNN,最后安装PyTorch。每个步骤都至关重要,确保了PyTorch的顺利安装和使用。

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

参考Win10 下安装 CUDA Toolkit ( for PyTorch )

1、查看电脑的CUDA驱动版本,确定cuda toolkit的兼容版本。
最好是让cuda toolkit的版本低于cuda的驱动版本
CUDA Toolkit 历史版本 中,选择 CUDA 11.0.3 进行下载。
在这里插入图片描述
在这里插入图片描述

2、下载和安装 cuda 11.0
在这里插入图片描述
成功之后,C盘下面的路径在这里插入图片描述

在这里插入图片描述
在这里插入图片描述

3、下载并安装与CUDA11.0版本兼容的CUDNN(为什么要安装cudNN)可以理解为是和cuda配套的东西,下载 cuDNN 需要注册,官网下载地址为 https://developer.nvidia.com/cudnn

在这里插入图片描述
在这里插入图片描述

所有内容复制到 cuda 11.0 安装目录(在C盘里面),即可完成 cuDNN 的安装。
在这里插入图片描述

在这里插入图片描述

以上步骤,缺一不可。

现在,才正式开始安装Pytorch.
可以直接安装pytorch\torchvision\cuda toolkit\ torch audio
在这里插入图片描述
在这里插入图片描述在这里插入图片描述

在这里插入图片描述
在这里插入图片描述
在这里插入图片描述

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值