cuda + cudnn + pytorch (win10)

win10 安装pytorch gpu 版

  • [查 显卡驱动](#查 显卡驱动)
  • [cuda 安装](#cuda 安装)
  • [cudnn 安装](#cudnn 安装)
  • 安装pytorch
  • 测试

查 显卡驱动

  • win + R : 输入cmd 唤出命令行;
  • 输入nvidia-smi
  • 关注驱动版本和cuda版本;
  • 后面安装的cuda版本不高于当前的cuda版本;

cuda 安装

  • 查看pytorch需要的cuda版本
  • cuda官网下载
  • 全部下一步自动安装。

cudnn 安装

  • 进入官网下载对应cuda版本的cudnn
  • 解压后,把解压出来的bin,include,lib文件夹内容,分别放入cuda对应的文件夹中。参考路径:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.6

安装pytorch

  • 在anaconda 中创建pytorch环境。conda create -n myEnvName python=3.9
  • 运行前面官网的安装指令。

测试

相关推荐
StarPrayers.3 小时前
(补)CNN 模型搭建与训练:PyTorch 实战 CIFAR10 任务的应用
人工智能·pytorch·cnn
Alex艾力的IT数字空间10 小时前
基于PyTorch和CuPy的GPU并行化遗传算法实现
数据结构·人工智能·pytorch·python·深度学习·算法·机器学习
chxin1401612 小时前
优化算法——动手学深度学习11
pytorch·python·深度学习
技术闲聊DD15 小时前
深度学习(14)-Pytorch torch 手册
人工智能·pytorch·深度学习
StarPrayers.15 小时前
CNN 模型搭建与训练:PyTorch 实战 CIFAR10 任务
人工智能·pytorch·cnn
言之。1 天前
Andrej Karpathy 演讲【PyTorch at Tesla】
人工智能·pytorch·python
DKunYu1 天前
PyTorch入门
人工智能·pytorch·python·深度学习
技术闲聊DD1 天前
深度学习(8)- PyTorch 数据处理与加载
人工智能·pytorch·深度学习
蓝博AI1 天前
基于卷积神经网络的香蕉成熟度识别系统,resnet50,vgg16,resnet34【pytorch框架,python代码】
人工智能·pytorch·python·神经网络·cnn
新子y2 天前
【小白笔记】关于 Python 类、初始化以及 PyTorch 数据处理的问题
pytorch·笔记·python