Ubuntu-torch.cuda.is_available() 的返回值为False
按照流程安装了GPU版本的pytorch,但是输入测试代码:
>>>import torch
>>>torch.cuda.is_available()
结果一直返回结果为False,如果提示是这个则代表GPU版本的pytorch无法正常使用。
结合网上其他的原因,检查了一遍,发现我的cuda和cudnn都安装成功了,版本也应该是匹配的。无奈只能回归到官网去寻找解决方案了
官网提示
注意到下面有对应的提示:先决条件
请按照此处描述的步骤进行操作:: https://github.com/pytorch/pytorch#from-source.
这里提示要先安装完CUDA和cudnn。接着安装依赖项,因为我是ubuntu18.04的版本,所以直接在终端创建好的虚拟环境下敲入代码:
conda install -c pytorch magma-cuda101#我之前安装的CUDA是10.1的版本
如下图所示(可以看到之前提示False的地方):
那么等待安装完之后再测试一下,结果就是True了!