记录下pytorch安装中一个比较隐蔽的坑,torch.cuda.is_available()返回false的一个原因

如题,想要安装GPU版,一开始从网上找了教程,显卡驱动、CUDA、cuDNN挨个更新安装,都选择了10.1版本,安装完后在anaconda中按照官网提示输入下面的代码,可以看到cudatoolkit即GPU版

conda install pytorch torchvision cudatoolkit=10.1 -c pytorch

然后发现基本上大一点的包都下载不下来,包括pytorch、cudatoolkit等等,遂又从网上找教程,加清华源,加了几个之后下载很快,下载下来也可以用,以为已经成功了,但是torch.cuda.is_available()返回就是False。

最开始以为是开始安装的版本不匹配,然后又尝试了10.2版本的CUDA,结果还是一模一样,最后输入conda list后终于发现了问题所在:

conda list中包的信息

可以看到第三列版本号前面的cpu,我的推测是清华源默认的是cpu版本,也有可能只有这一个版本,这个还有待大家发掘。我是科学上网后直接输入官网给出的代码,顺利安装后再次输入torch.cuda.is_available(),这次果然返回的是True。

记录下来供大家参考。

你可能感兴趣的:(pytorch,cuda,PyTorch入门)