torch.cuda.is_available()false的原因之一

问题描述

使用pytorch训练数据集时GPU一直没有被调用，输入torch.cuda.is_available() 返回值为false

解决方法

如果你也是使用官网安装方法、确定cuda和torch版本对应的话，

大概率是pytorch下载版本为cpu版本（并不是一开始选择不对，我也是一开始如上选择）但是后面下载中断了一次 重新源代码重新下载后输入

conda list

发现下载安装的为cpu版本

所以提供两个链接进行本地安装
清华源
如果云端下载安装输入命令行时将cuda版本改掉
如果下载到本地

conda install --use-local your-pkg-name
例如 conda install --use-local pytorch-0.4.0-py35_cuda8.0.61_cudnn7.1.2_1.tar.bz2

官方源

pip install xxx