pytorch (GPU) 安装 全部过程(踩得坑好多)(win10)

本文详述了在Windows 10上安装PyTorch GPU版本的全过程,包括确定CUDA版本、安装CUDA和CUDNN、配置虚拟环境以及解决安装过程中遇到的各种问题,如驱动兼容性、算力不足等。同时强调了在Anaconda虚拟环境中安装PyTorch的好处和步骤。
摘要由CSDN通过智能技术生成

惨痛史:

自己一开始由于没有什么了解,就随便网上找了一个教程开始弄,也没有在意是否匹配自己的显卡驱动情况,就安装了,最开始受罪的是我的Ubuntu,哎,这个就不提了,都是泪,之后再在windows10 上安装cuda、cudnn等也是一头雾水,各种版本问题啊、下载问题啊、虚拟环境问题啊、BERT支持问题啊,导致按了又重装一遍,一开始安装的时候NVIDIA驱动都被我搞没了,安装完成之后用几天等到不匹配的时候才发现这个问题,心态都有点要崩了(像我这么积极的人 哈哈 ),昨天开始全部重新安装,顺便写下全部过程,应该是踩了不少坑,希望可以对后来人有所帮助。

写在开头

大家先要了解自己的驱动支持版本,我是在昨天更新驱动之后才可以支持cuda10.1的,要不然之前一直最高只支持9.1版本的,那么如何查看本地驱动支持的CUDA版本呢?
操作:打开NVIDIA驱动程序,选择导航栏帮助-系统信息,选择组件,查看动态链接库文件NVCUDA.DLL 后面会写最高支持的CUDA版本哦
在这里插入图片描述
这里再新增一点(菜鸟的我又踩到坑了。。。):虽然驱动支持cuda10.1的,但是后面在跑模型的时候出现本机的算力不足的问题,也就是硬件无法支持cuda的高版本,最后我卸载cuda10.1的,然后重装了cuda9.0的,这个问题就解决了,笔记本还是有点菜啊(920M 对应算力3.5)
这里附一张显卡算力表:
在这里插入图片描述
下面带一张cuda对应需求算力:
在这里插入图片描述

下载对应CUDA版本

这里由于官网找得比较麻烦,大家可以通过点击下面链接找到较为全面的CUDA版本(链接地址是:https://developer.nvidia.com/cuda-toolkit-archive)
CUDA各版本官方下载地址

安装CUDA

下载之后就安装吧
这里顺便提一下,因为CUD

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值