admin管理员组

文章数量:1577502

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/deploy/cuda-compatibility/#binary-compatibility

安装方法：

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

 

本文标签： 解决办法CUDAtorchfalseisavailable

版权声明：本文标题：torch.cuda.is_available()返回false的解决办法 内容由热心网友自发贡献，该文观点仅代表作者本人， 转载请联系作者并注明出处：https://www.elefans.com/dongtai/1725726835a1039062.html， 本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌抄袭侵权/违法违规的内容，一经查实，本站将立刻删除。