当前位置:首页 » 《随便一记》 » 正文

深度学习笔记1:“安装CUDA、Pytorch、调试GPU环境“方法以及常见误区_yuhongchuancsu的博客

16 人参与  2021年12月26日 12:00  分类 : 《随便一记》  评论

点击全文阅读


因为我的电脑是在18年买的,比较旧(GTX1050),安装的Python版本又很新(3.8.8),接下来面临的主要问题就是版本兼容问题。对于安装方法,网上总结的很全了,这里总结一下重要的安装要点,也是新手容易困惑的地方

误区1:CUDA安装版本上限

查找网上CUDA安装方法,很多文章都会通过“NAVIDA控制面板→帮助→系统信息→组件”查找到CUDA的版本,并且认定这是和系统兼容的版本(即:安装的版本上限)。我认为这是错误的。

我认为这是错误的原因:我的显卡是18年出的GTX1050,按照上述方法,我开始查到的CUDA版本号是9.1。后来通过试验发现:即使我安装了19年出的CUDA10.2,也是兼容的。因此我认为:按照上述方法给出的CUDA版本号并不是安装上限,而是你曾经安装过的CUDA版本号或者说你显卡推出年份的CUDA版本号。

如果安装高于这个的CUDA版本,是没有问题的。以我自己为例,至少GTX1050是可以安装10.2的。

我为什么特意强调了这一误区呢?这一误区导致我开始认为要么安装Python老版本,要么换一个新显卡,否则无法调用GPU环境了(悲)。因为我的Python版本很新,导致对Pytorch和CUDA的安装版本要求也比较新,老的CUDA无法和Pytorch、Python同时兼容。如果能够认识到这个误区,就可以根据自己需求下载这三个软件了,不用再担心“显卡过旧和CUDA不兼容”的问题。

方法1:最快查询CUDA、Pytorch、Python兼容版本的方法,少走很多弯路

即使电脑成功安装了CUDA和Pytorch,如果版本不兼容,也无法调用GPU环境。我自己就在这步花费了很多时间。

有文章通过Pytorch官网的pip命令安装Pytorch库。对于更快的方法,我直接上结论:通过pytorch下载界面的文件名来看(下载界面的链接https://download.pytorch.org/whl/torch_stable.html)

如上图,whl文件的文件名直接告诉了我们兼容的CUDA、pytorch和python版本号,选择合适的文件下载即可。

这样还有一个好处:如果通过官网的pip安装命令下载pytorch,可能需要梯子,且下载速度很慢,进而导致pytorch无法正常安装。如果通过浏览器先下载好whl文件,再安装,速度就很快。对于whl文件的安装方法,参考https://jingyan.baidu.com/article/08b6a5917ed44214a909225c.html

方法2:“NIVIDA安装程序失败”处理方法

在这里插入图片描述

(1)在安装CUDA之前,点开控制面板,卸载所有NIVIDA公司的产品,保留框里的两个。然后用360或者电脑管家清理卸载残留,清理注册表。

(2)  进入C:\Program Files,删除NIVIDA  Corporation文件夹。如果提示文件正在运行,先用任务管理器中止NIVDA组件运行再删除,或者利用电脑管家的文件粉碎机强行删除。

(3)安装时,不要勾选“Visual Studio Integration”组件

在这里插入图片描述

(4)CUDA安装后,要进行环境变量配置和CUDNN的文件配置,推荐一篇博客:win10+CUDA安装及环境配置_Henry的博客-CSDN博客_cuda配置环境变量

方法3:其它问题的补充

(1)调用GPU环境的代码

import torch
tensor = torch.rand(3,4)
print(f"Shape of tensor: {tensor.shape}")
print(f"Datatype of tensor: {tensor.dtype}")
print(f"Device tensor is stored on: {tensor.device}")
if torch.cuda.is_available():
    tensor = tensor.to('cuda')
    print(f"Device tensor is stored on: {tensor.device}")
    print(torch.cuda.is_available())#如果结果为True,说明GPU环境可以调用

(2)GPU环境下,初次运行深度学习代码,可能出现虚拟内存不足的问题,需要为磁盘开辟虚拟内存

(3)"CUDA out of memory" 解决方法:调小图片尺寸、降低Batch_size、令num_workers=0、令pinmemory=0、或者买个显卡(玩笑 


点击全文阅读


本文链接:http://zhangshiyu.com/post/32148.html

安装  版本  方法  
<< 上一篇 下一篇 >>

  • 评论(0)
  • 赞助本站

◎欢迎参与讨论,请在这里发表您的看法、交流您的观点。

最新文章

  • 变了心的女人,再爱也不能要听书_乐瑶祁知白闻言全新_小说后续在线阅读_无删减免费完结_
  • 尚梦寒烟的我不寻龙脉后,自称能与山灵沟通的实习生一夜白头许云天秦若曦全书在线
  • 何人葬我忘川岸(凤灼华厉沧溟)(何人葬我忘川岸)在线畅读+开篇大作+前传
  • 丈夫让我妈捐骨髓,救他女助理的妈(周云红贺知远)全书免费_(周云红贺知远)丈夫让我妈捐骨髓,救他女助理的妈后续(周云红贺知远)
  • 回首桃花已盛开(方梨初陆津年)_回首桃花已盛开(方梨初陆津年)
  • 死后三年,老公逼我捐肾给白月光(傅谨初沈云苒)_死后三年,老公逼我捐肾给白月光
  • 王妃重生后,战王抱着崽来认错(沈意宋云祈)全书免费_(沈意宋云祈)王妃重生后,战王抱着崽来认错后续(沈意宋云祈)
  • 江晏屿许妍惜小说(月亮为茉莉祈祷)序章+全章阅读(江晏屿许妍惜)最新章节
  • (复仇千金有亿点强,大佬卑微求爱)复仇千金有亿点强,大佬卑微求爱小说(安若曦萧渊)小说全集速览无广告高口碑小说
  • 学术妲己嫁人怀孕,科研大佬悔疯了最新阅读_顾砚柳白萧墨潜力榜_小说后续在线阅读_无删减免费完结_
  • 阮糖宙神被我渣过的男主全员疯批小说完本+番外(被我渣过的男主全员疯批)畅享阅读
  • 夏南兮盛简州小说(那年盛夏爱无风)全文+后续(夏南兮盛简州)无套路阅读

    关于我们 | 我要投稿 | 免责申明

    Copyright © 2020-2022 ZhangShiYu.com Rights Reserved.豫ICP备2022013469号-1