最近在安装pytorch时极为恼火,明明电脑有GPU且pytorch已经装好,但是torch.cuda.is_available()一直是返回false。在网上搜集了一堆解决办法,最终摸索了两天后解决了这个问题。为避免下次遇到此问题,特此做个解决方法记录。
当出现torch.cuda.is_available()返回false的情况时解决办法
- 首先在自己创建的用于安装pytorch的虚拟环境中,输入
conda list
查看从官网下载的pytorch是CPU版本的还是GPU版本的。 - 若发现自己下载的pytorch为CPU版本的,则将此pytorch卸载,去官网使用PIP安装方式安装新的pytorch,记住千万,千万,千万不要用conda的安装方式,因为它下载的一直都是CPU版本的。
- 若自己下载的确实为GPU版本的,但是仍旧出现torch.cuda.is_available()返回false的情况,则可能是版本不匹配的问题。