首页 最新 热门 推荐

  • 首页
  • 最新
  • 热门
  • 推荐

torch.cuda.is_available()返回false的解决办法

  • 25-03-03 22:41
  • 4710
  • 11315
blog.csdn.net

1、问题

安装好torch  gpu版本之后,torch.cuda.is_available()一直返回False;   但是执行 torch.backends.cudnn.enabled is TRUE.

执行Nvidia-smi 命令没有报错,能够显示驱动信息;

在网上,搜的解决方法:执行命令:

sudo apt-get install nvidia-cuda-toolkit

仍然报错。

 

2、问题分析

尝试各种办法之后,还是依然返回False,正常如果安装正确的话,是返回TRUE的,问题就在于版本的问题,要么是显卡驱动版本的不匹配,要么是安装的包不匹配。

3、解决办法:

(1)方法一:更新显卡驱动,这种方法,风险比较大,操作比较麻烦,个人不建议;

(2)方法二:找到对应的 cudatoolkit对应的版本进行安装:具体各个驱动支持的版本对应,如下:

https://docs.nvidia.com/deploy/cuda-compatibility/#binary-compatibility

安装方法:

 conda install pytorch torchvision cudatoolkit=xxx(选择对应的版本) -c pytorch

 

文章知识点与官方知识档案匹配,可进一步学习相关知识
CUDA入门技能树首页概览4119 人正在系统学习中
注:本文转载自blog.csdn.net的weixin_41596463的文章"https://blog.csdn.net/weixin_41596463/article/details/102525414"。版权归原作者所有,此博客不拥有其著作权,亦不承担相应法律责任。如有侵权,请联系我们删除。
复制链接
复制链接
相关推荐
发表评论
登录后才能发表评论和回复 注册

/ 登录

评论记录:

未查询到任何数据!
回复评论:

分类栏目

后端 (14832) 前端 (14280) 移动开发 (3760) 编程语言 (3851) Java (3904) Python (3298) 人工智能 (10119) AIGC (2810) 大数据 (3499) 数据库 (3945) 数据结构与算法 (3757) 音视频 (2669) 云原生 (3145) 云平台 (2965) 前沿技术 (2993) 开源 (2160) 小程序 (2860) 运维 (2533) 服务器 (2698) 操作系统 (2325) 硬件开发 (2491) 嵌入式 (2955) 微软技术 (2769) 软件工程 (2056) 测试 (2865) 网络空间安全 (2948) 网络与通信 (2797) 用户体验设计 (2592) 学习和成长 (2593) 搜索 (2744) 开发工具 (7108) 游戏 (2829) HarmonyOS (2935) 区块链 (2782) 数学 (3112) 3C硬件 (2759) 资讯 (2909) Android (4709) iOS (1850) 代码人生 (3043) 阅读 (2841)

热门文章

101
推荐
关于我们 隐私政策 免责声明 联系我们
Copyright © 2020-2025 蚁人论坛 (iYenn.com) All Rights Reserved.
Scroll to Top