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
  • 运行前面官网的安装指令。

测试

相关推荐
努力还债的学术吗喽19 小时前
PyTorch nn.Linear 终极详解:从零理解线性层的一切(含可视化+完整代码)
人工智能·pytorch·python·深度学习·基础组件·线性层·nn.linear
铁手飞鹰1 天前
从零复现论文:深度学习域适应1
linux·pytorch·python·深度学习·ubuntu·ai·迁移学习
拾贰_C1 天前
【pycharm---pytorch】pycharm配置以及pytorch学习
pytorch·学习·pycharm
西猫雷婶1 天前
pytorch基本运算-torch.normal()函数生成的随机数据添加噪声
人工智能·pytorch·python·深度学习·学习·线性代数·机器学习
Yolo566Q1 天前
基于PyTorch深度学习遥感影像地物分类与目标检测、分割及遥感影像问题深度学习优化实践技术应用
pytorch·深度学习·分类
源码部署22 天前
Pytorch零基础入门到项目实战
pytorch
小林z2 天前
Pytorch零基础入门到项目实战
pytorch
Sunhen_Qiletian2 天前
用PyTorch实现CBOW模型:从原理到实战的Word2Vec入门指南
人工智能·pytorch·word2vec
高-老师2 天前
基于PyTorch深度学习无人机遥感影像目标检测、地物分类及语义分割实践技术应用
pytorch·深度学习·无人机
深度学习机器2 天前
解密vLLM:基于nano-vllm源码剖析其推理加速之道
pytorch·llm·nvidia