cuda.isavailableture为什么还是cuo

cuda.isavailableture为什么还是cuo,第1张

一. 出现torch.cuda.is_available为False的原因

安装的版本问题

可能安装了PyTorch的CPU版本,它所安装的Pytorch是不带CUDA的,代码处理图像时没有用GPU而只用了CPU

检查GPU是否支持CUDA问题

查找自己的GPU是否支持cuda,绝大多数GPU都是支持cuda的.

(1)方法一:ctrl + alt+ delete 进入任务管理器,选择“性能”,查看GPU是否支持CUDA

(2)方法二:桌面右键查看是否有NVIDIA控制面板

GPU不支持cuda

如何成功在没有nvidia驱动的win10电脑里安装CUDA和pytorch-gpu

参考这个链接(我还没安装成功)

https://blog.csdn.net/weixin_41194129/article/details/107475509?utm_term=gpu%E6%B2%A1%E6%9C%89cuda&utm_medium=distribute.pc_aggpage_search_result.none-task-blog-2allsobaiduweb~default-2-107475509&spm=3001.4430

检查显卡驱动版本问题

(1)插卡本机的NVIDIA显卡驱动,

目的:现在NVIDIA控制面板里找到显卡驱动版本支持什么cuda版本,如本机cuda9.2

方法:NVIDIA控制面板 → 帮助 → 系统信息 → 组件 → 3D设置 → NVIDIA.DLL中查看支持版本

(2)查NVIDIA显卡驱动与CUDA Toolkit版本对照表,下方链接:

https://blog.csdn.net/heiheiya/article/details/103868478

(3)根据对照表选择合适的CUDA Toolkit ,如本机

二. 出现torch.cuda.is_available为False的影响

(1) 运行时间长

(2)输出图像效果极差

三. 出现torch.cuda.is_available为False解决方法

卸载Pytorch

如果当初是用conda install 命令安装的pytorch,则用第一种方法,如果是用pip install命令安装的,则用第二种方法。或者两种方法都试一下

(1)使用conda卸载Pytorch

conda uninstall pytorch

conda uninstall libtorch

(2)使用pip卸载Pytorch

pip uninstall torch

windows下直接输入nvidia-smi显示不是内部或外部命令也不是可运行的程序

(1)cmd窗口改变路径到C:\Program Files\NVIDIA Corporation\NVSMI

改变路径方法:CD /D C:\Program Files\NVIDIA Corporation\NVSMI

(2)在上面的路径上,输入nvidia-smi查看NVIDIA显卡驱动drive version

要求:drive version 必须大于NVIDIA版本需求

比如:本机为cuda9.2 驱动版本要求大于或等于396.26,即drive version要大于或等于396.26.

选择合适的cuda 和cudatoolkit版本下载。

可以下载好cuda和cudatoolkit版本安装包,用命令conda --use–local 包名 进行安装

在Windows10下,Git clone https://github.com/ultralytics/yolov5.git

然后运行:

问题原因 :win10下特有的问题,win10的默认编码方式是utf-8,而Python读入文本文件的默认编码方式是gbk

解决方式 :找到read()函数,显示指定编码方式。根据报错信息,read()在 dist.py文件的406行 ,所以打开该文件,加入“encoding="utf-8”,如下图所示:

我的 1080Ti 算力为6.1 ,从下面的三个表可以分析,还可以跟着PyTorch升级再战几年!

YOLOv5的当前(3月28日)PyTorch版本:YOLOv5

欢迎分享,转载请注明来源:内存溢出

原文地址:https://54852.com/tougao/9359231.html

(0)
打赏 微信扫一扫微信扫一扫 支付宝扫一扫支付宝扫一扫
上一篇 2023-04-27
下一篇2023-04-27

发表评论

登录后才能评论

评论列表(0条)

    保存