当前位置: 首页 > news >正文

安装CUDA、CUDNN、pytorch

2.1 安装CUDA

2.1.1找到CUDA的版本号

2.1.1.1 屏幕的左下角,找到NVIDIA控制面板的图标,如下图所示:

图2.1.1.1  NVIDIA控制面板图标

2.1.1.2 打开NVIDIA控制面板,找到系统信息,如下图所示:

图2.1.1.2 系统信息

2.1.1.3 在组件里,查看CUDA的版本号,如下图所示:

图2.1.1.3  查看CUDA版本号

2.1.2 安装CUDA

如上图所示,CUDA的版本号是12.2.148,到nvidia的官方网站https://developer.nvidia.com/cuda-toolkit-archive下载对应的版本,如下图所示:

图2.1.2.1 找到CUDA版本

点击链接,进入,按下图所示选择并下载

图2.1.2.1 下载CUDA

下载后,安装即可。

2.2 安装cuDNN

2.2.1 找到cuDNN的版本号

   打开网站https://developer.nvidia.com/rdp/cudnn-archive,找到CUDA的版本号是12.2.148的cuDNN的版本号是V8.9.3(月年都是July 2023),如下图所示:

图2.2.1.1 cuDNN版本号

点击后,选择Windows的版本,如下图所示:

图2.2.1.2  cuDNN Windows版本

点击后,会提示登录,没有账号需要注册,如下图所示:

图2.2.1.3  NVIDIA登录

2.2.2 安装cuDNN

1)下载后,解压,如下图所示:

​图2.2.2.1  cuDNN解压后文件夹

2)找到CUDA的安装路径,一般在C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.2,如下图所示:

图2.2.2.2  CUDA安装路径

        将cuDNN解压后的bin、include、lib三个文件夹,复制粘贴到CUDA的安装文件里。 

2.3 安装PyTorch

   1)打开PyTorch的官网PyTorch,并按图选择, 如下图所示:

图2.3.1  PyTorch网站

   2)在应用中找到并打开Anaconda Prompt,输入conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia如下图所示:

图2.3.2 运行安装pytorch指令

安装成功,如下图所示:

图2.3.3 pytorch安装成功

  1. 如果安装失败,选择CPU,如下图所示:

图2.3.4  pytorch安装选择CPU

4)打开Anaconda Prompt,输入conda install pytorch torchvision torchaudio cpuonly -c pytorch


http://www.mrgr.cn/news/4272.html

相关文章:

  • 【YOLO家族】Scaled-YOLOv4, Optimal Speed and Accuracy of Object Detection,CVPR 2021
  • 一起学习LeetCode热题100道(52/100)
  • MinIO的性能优化
  • 无人共享自助棋牌室台球室茶室系统源码Java一体化解决方案
  • 软考高级考完了,怎么评职称?
  • 数据治理—数据标准管理
  • XSS-Jquery.html()+DOM破坏
  • 让AI转化为生产力,Dify 私有化部署 + 接入Ollama部署本地模型。
  • 公开数据库汇总及下载(1)-TCGA
  • 代码随想录算法训练营day48|单调栈part01
  • Java语言程序设计基础篇_编程练习题***16.31(游戏:四子连)
  • Redis补充
  • [机器学习]--线性回归算法
  • 微信答题小程序产品研发-后端开发
  • 封装了一个iOS评论弹窗
  • 鸿蒙Harmony实战:自动化测试开发原理
  • LearnOpenGL——HDR、Bloom学习笔记
  • Spring Mybatis拦截器配合logback打印完整sql语句
  • EGL函数翻译--eglMakeCurrent
  • NC 矩阵元素查找