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

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