Pytorch(gpu),cuda,cudnn安装

  • 第一步:查看自己电脑显卡支持的CUDA版本
  • 第二步:查看pytorch对应cuda版本
  • 第三步:CUDA下载安装
    • 遇到问题:安装完之后法找到nvcc命令
  • 第四步:cudnn下载安装
    • 添加环境变量(比较玄学,有的需要,有的不需要),我的不需要,所以图是偷的
    • 遇到这种情况不要慌,看说明需要在前面加上 `.\deviceQuery.exe`和`.\bandwidthTest.exe`,收获两个pass代表安装完成
  • 第五步:安装pytorch
  • 第六步:验证pytorch测试GPU是否安装成功

第一步:查看自己电脑显卡支持的CUDA版本

Nvidia-smi


注意:这里的11.7是电脑所能支持的最高版本

第二步:查看pytorch对应cuda版本

Pytorch官网查看对应版本关系
标注的地方为查看历史版本

注意区分CPU版本 和 CUDA版本下错版本会出现GPU 返回 False

第三步:CUDA下载安装

我使用的是 pytorch 1.12.1 版本,所以需要 cuda 11.6 版本
提醒:cudatoolkit就是cuda
CUDA下载官网

找到 11.6 版本

下载到指定的文件夹,运行exe文件
注意!!!:这个文件夹是临时文件夹,安装重启完之后会自动删除,所以后面文件安装不能在此文件夹内




这里的目录不能设置在刚刚的临时文件里面,不然白给



安装完成之后,查看系统环境变量
推荐两种方法:
第一种:按快捷键win+R后,输入sysdm.cpl

第二种:windowns搜索框搜索 环境变量

查看环境变量

重启电脑,重启完之后,cmd打开控制台,输入nvcc -V查看安装是否成功
注意:是大写 V

彩蛋:不想重启的 让环境变量立即生效的方法:
cmd进入控制台,输入set PATH=c,回车
输入 echo %PATH%,回车,返回 c
关闭窗口,之后再次进入控制台,输入echo %PATH%
这样新添加的环境变量已经生效了。
输入nvcc -V验证是否成功

遇到问题:安装完之后法找到nvcc命令

打开path路径,手动添加环境变量

每个人路径不一样
D:\python\Cuda\CUDA\bin
D:\python\Cuda\CUDA\libnvvp

第四步:cudnn下载安装

进入cudnn官网,你需要注册一个cudnn官网账号,有的直接登录即可

登陆完之后会出现这个页面

找CUDA 11.6 对应版本的cudnn,这里我选择的是 8.4.0

下载解压完之后,复制三个文件到指定文件夹,直接ctrl+c,ctrl+v,不用一个一个对应

添加环境变量(比较玄学,有的需要,有的不需要),我的不需要,所以图是偷的

需要的话,把刚刚三个文件夹的路径添加到path中

验证是否安装成功
进入cuda安装路径,D:\python\Cuda\CUDA\extras\demo_suite,找到deviceQuery.exebandwidthTest.exe

进入此目录下的控制台,运行deviceQuery.exebandwidthTest.exe

遇到这种情况不要慌,看说明需要在前面加上 .\deviceQuery.exe.\bandwidthTest.exe,收获两个pass代表安装完成

第五步:安装pytorch

不用使用anaconda安装,太慢了,还总是失败
这里使用pip安装,版本号选择参考第二步
我这里只选择 pytorchtorchvision

pip3 install -i https://pypi.tuna.tsinghua.edu.cn/simple torch==1.12.1 torchvision==0.13.1 -f https://download.pytorch.org/whl/torch_stable.html

静等安装成功

第六步:验证pytorch测试GPU是否安装成功

验证脚本是嫖的

import torch
flag = torch.cuda.is_available()
print(flag) # 返回true为安装成功ngpu= 1
# Decide which device we want to run on
device = torch.device("cuda:0" if (torch.cuda.is_available() and ngpu > 0) else "cpu")
print(device)
print(torch.cuda.get_device_name(0))
print(torch.rand(3,3).cuda())

Pytorch(gpu),cuda,cudnn安装相关推荐

  1. 真实机下 ubuntu 18.04 安装GPU +CUDA+cuDNN 以及其版本选择(亲测非常实用)【转】...

    本文转载自:https://blog.csdn.net/u010801439/article/details/80483036 ubuntu 18.04 安装GPU +CUDA+cuDNN : 目前, ...

  2. linux下cuda cudnn安装 没有权限的安装

    linux下cuda cudnn安装 cuda 10.2:https://developer.nvidia.com/cuda-10.2-download-archive?target_os=Linux ...

  3. Ubuntu系统---安NVIDIA 驱动后 CUDA+cuDNN 安装

    Ubuntu系统---安NVIDIA 驱动后  CUDA+cuDNN 安装 --------------------------------------------@20190726--------- ...

  4. Win10 Anaconda下TensorFlow-GPU环境搭建详细教程(包含CUDA+cuDNN安装过程)(转载)...

    win7(win10也适用)系统安装GPU/CPU版tensorflow Win10 Anaconda下TensorFlow-GPU环境搭建详细教程(包含CUDA+cuDNN安装过程) 目录 2.配置 ...

  5. Win10 + ubuntu16双系统安装以及cuda,cudnn安装

    Win10 + ubuntu16双系统安装以及cuda,cudnn安装 其实这些东西我已经弄过好多次了但是每一次都遇到坑,所以记录一下.仅供参考 1.windows安装 这一步较为简单,注意安装系统到 ...

  6. GPU, CUDA,cuDNN三者的关系总结

    GPU, CUDA,cuDNN三者的关系总结 CPU 和 GPU cpu和gpu结构展示 GPU(图像处理器,Graphics Processing Unit)和CPU(中央处理器,Central P ...

  7. PyTorch+CUDA+cudnn安装

    1.安装CUDA 先查看一下电脑显卡对应的CUDA驱动版本 上图可以看到CUDA驱动版本为11.1.114,那就下载11.1版本的CUDA好了.(这里应该也不是必须对应,只是说有可能下载低版本的会有问 ...

  8. ubuntu查看cudnn是否安装成功_深度学习之目标检测系列(0) -ubuntu18.04+RTX2080Ti+cuda+cudnn安装...

    ubuntu18.04使用cudnn的过程中会遇见很多问题,而且搜索很多文章发现比较杂乱,我这里总结一下,方便小伙伴少走几步坑. 我的服务器硬件配置信息 (2019-01购买) CPU: 英特尔(in ...

  9. conda cudnn版本升级_ubuntu18.04系统下NVIDIA显卡驱动+cuda+cudnn安装+conda虚拟环境配置...

    深度学习的显卡环境配置是一个非常麻烦的问题,理论上这是一个深度学习开发必须要面临的问题,那么教程就应该很多,后人学习起来的时候应该会很容易,实际上却是教程良莠不齐,而且很多教程往往是针对特定显卡的,但 ...

  10. Tensorflow-gpu+Cuda+cuDNN安装(详细且包含失败解决教程)

    WINDOWS系统查看nvidia显卡和CUDA的版本号 Win+r+cmd后输入nvidia-smi 主要是看Driver Version 462.31这个值. 2.根据Driver Version ...

最新文章

  1. matlab最小二乘法参数辨识,基于最小二乘法的MATLAB辨识仿真实验
  2. Vue自定义指令实现下拉加载:v-loadmore
  3. java同步变异步框架_java-如何使用Spring配置异步和同步事件发布者
  4. mysql 存储地理信息_使用Amazon Aurora存储和处理地理信息数据
  5. CEO 赠书 | 讲述创新背后不为人知的故事,改变世界的 Ta 们做对了什么?
  6. java 将查询到的值 存下_MyBatis3.x 从入门到精通(六)查询
  7. Redis模式匹配删除key
  8. 电磁波传播相位是否会变化_相位常数β与波数k是一回事吗?
  9. “ == “运算符与equals()方法的区别
  10. 第23天:js-数据类型转换
  11. 【Scala】Scala语言的介绍以及循环的定义(while,for,break,九九乘法表的计算代码)
  12. 用html打开dbf文件怎么打开,dbf格式文件是什么 dbf文件用什么软件打开
  13. JAVA WEB 对返回数据进行按中文名称首字母A~Z升序排序
  14. DOM、Window操作
  15. 2019.5.25 Noip模拟测试2 T1题解
  16. 20170908在线编程之圆周上问两点间的距离问题
  17. 2021-06-22列表样式与背景图片
  18. java jvm学习笔记三(class文件检验器)
  19. 基于51单片机中文汉字LCD12864滚动显示屏仿真(源码+仿真+全套资料)
  20. 本科毕业设计论文答辩ppt模板

热门文章

  1. 如何应对redis缓存的击穿、穿透和雪崩
  2. Oracle 使用序列创建自增字段
  3. 尚医通(九)数据字典模块前后端 | EasyExcel
  4. 用C语言写一个停车场管理系统代码
  5. Flink_企业级风控平台架构建设_01
  6. 【毕业设计_课程设计】基于神经网络学习的在线纸币识别【源码+论文】
  7. 深度学习第四次培训(SVM算法)
  8. SQL入门SQL基本介绍
  9. 期货法律法规重点笔记2
  10. 怎样关闭任务栏的Microsoft资讯