Pytorch1.10+CUDA10.2+cudnn安装成功之路

本文记录了在安装Pytorch 1.10和CUDA 10.2过程中遇到的挑战,包括CPU版本Pytorch、CUDA版本与Pytorch不兼容、环境变量配置错误等问题。通过调整CUDA版本、修复Python环境和正确安装whl文件,最终成功使CUDA支持变为true。
摘要由CSDN通过智能技术生成

最近跑别人论文里的代码,要用到BERT模型,但是本人没有.深度学习神经网络方面的基础,刚开始模型什么都看不懂,后来看一点神经网络的一些基础知识点,也没看完就看不下去了,急着想把代码跑一跑,然后就死在第一步了,安装环境!首先说下我的情况,我的代码里本来有import torch,我刚开始不知道这个torch是什么,就以为普通的包,就直接在pycharm引入,果不其然失败了,我就去网上搜了一个下载方式,后来我才知道我下载的是cpu版本的pytorch 1.10。本以为下载好了,我就去运行代码了,结果预训练也需要下载模型和数据集,我真的什么都不知道我就又去下载预训练模型需要的东西,终于下好了以为这下可以跑代码了吧,结果报错,就是下图:错误提示

 我就一脸懵又去搜了下这个错误,我才发现,一个模型能跑起来要依赖的东西还挺多的,我是真滴菜!

pytorch、显卡、显卡驱动、cuda版本是如何对应的 - 简书 (jianshu.com)

然后我就去搜了博客按着博客一步一步安装CUDA和cudnn,参考博客如下:

  • 6
    点赞
  • 19
    收藏
    觉得还不错? 一键收藏
  • 2
    评论
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值