文章详情

短信预约-IT技能 免费直播动态提醒

请输入下面的图形验证码

提交验证

短信预约提醒成功

python windows安装cuda+cudnn+pytorch教程

2023-05-19 08:55

关注

安装cuda 我强调下 这个需要注意版本问题的

注意 (个人的想法,安装思路,仅供参考)

pytorch 需要注意这个现在支持的版本.根据这个支持的版本去下载对应的cuda和cudnn 

应为你 pytorch 的版本对不上 你cuda装上了也不行的. 到时候检测运行的时候直接就False.

我现在安装的pytorch 支持 cuda的版本是 11.3. 所以你们安装的时候需要注意. 应该是可以用低版本的安装的. (指的是cuda). 切记,切记,切记,

https://pytorch.org/ 这是 pytorch的官网.

简单的说

以下出现的所有cuda和cudnn的版本都是作为安装教程使用,具体版本根据情况而定.

以下出现的所有cuda和cudnn的版本都是作为安装教程使用,具体版本根据情况而定.

以下出现的所有cuda和cudnn的版本都是作为安装教程使用,具体版本根据情况而定.

图片只是记录一个安装流程.

先校验英伟达显卡版本.(后续切勿自动更新这个)

对应显卡版本进行下载对应的cuda版本 (也可以降低版本安装)

这是网址 https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html

找到自己合适的版本即可.

安装的版本 不能超过pytorch限制的版本

简单的说下这里的逻辑:

下载安装 cudaa

下载链接 https://developer.nvidia.com/cuda-toolkit-archive

下载自己合适的版本.

我的是windows10 我选择本地安装.

进行安装.

文件存放的地方

需要安装的模块

模块安装的位置

安装完成之后 查看下自己环境变量

这样基本上就ok了

简单的测试下

打开cmd终端 ,执行命令

nvcc -Vnvcc --version

这样就ok了

下载 cudnn

下载网址 https://developer.nvidia.com/rdp/cudnn-archive

下载合适自己的版本

将cudnn文件进行解压 放到cuda的目录下.

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.4 这里我的路径 每个人的路径可能不一样.

测试使用下.

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.4\extras\demo_suite 找到这个demo_suite 这个文件夹

然后调用这个exe.

效果如下就基本ok了

官网下载安装 pytorch https://pytorch.org/

import torch
print(f'torch -v {torch.__version__}')
print(f'cuda is_available {torch.cuda.is_available()}')

结果:

torch -v 1.10.1+cu113

cuda is_available True

总结

以上为个人经验,希望能给大家一个参考,也希望大家多多支持编程网。

阅读原文内容投诉

免责声明:

① 本站未注明“稿件来源”的信息均来自网络整理。其文字、图片和音视频稿件的所属权归原作者所有。本站收集整理出于非商业性的教育和科研之目的,并不意味着本站赞同其观点或证实其内容的真实性。仅作为临时的测试数据,供内部测试之用。本站并未授权任何人以任何方式主动获取本站任何信息。

② 本站未注明“稿件来源”的临时测试数据将在测试完成后最终做删除处理。有问题或投稿请发送至: 邮箱/279061341@qq.com QQ/279061341

软考中级精品资料免费领

  • 历年真题答案解析
  • 备考技巧名师总结
  • 高频考点精准押题
  • 2024年上半年信息系统项目管理师第二批次真题及答案解析(完整版)

    难度     813人已做
    查看
  • 【考后总结】2024年5月26日信息系统项目管理师第2批次考情分析

    难度     354人已做
    查看
  • 【考后总结】2024年5月25日信息系统项目管理师第1批次考情分析

    难度     318人已做
    查看
  • 2024年上半年软考高项第一、二批次真题考点汇总(完整版)

    难度     435人已做
    查看
  • 2024年上半年系统架构设计师考试综合知识真题

    难度     224人已做
    查看

相关文章

发现更多好内容

猜你喜欢

AI推送时光机
位置:首页-资讯-后端开发
咦!没有更多了?去看看其它编程学习网 内容吧
首页课程
资料下载
问答资讯