安装pytorch
总会遇到这样那样的问题,导致要么import torch
出错,要么torch.cuda.is_available()
输出False
。今天把我遇到的所有问题整理了一个简单的总结,仅供参考。
1 显卡
首先确保你的显卡支持CUDA
。 (不展开,自己面对百度)
2 驱动
驱动最好是升级到最新。我是下载了这个:GeForce Experience,然后升级里面的GeForce Game Ready
到最新就可以。
3 CPU or GPU
如果以上两个都没有问题,可以通过Anaconda Prompt
和conda activate 环境名
进入你的环境,然后检查你安装的pytorch
是否有cpuonly
到conda list
,因为长安装时有gpu
和gpu
版本。
如果有cpuonly
字,可以用conda uninstall cpuonly
(或pip
)卸载。
如果不适合你,建议去官网重新下载pytorch
。通过pip wheel
命令下载一般是没有问题的。这就是我解决它的方法。
4 版本
如果你之前安装并成功调用了cuda
,但是你的pytorch
版本由于某种原因发生了变化(有些包是捆绑下载的,下载后会自动更改pytorch
版本),你就不能再调用GPU
了。这个时候最好的解决办法是把pytorch
安装回原来的版本。
如果你太过于着急,已经把cuda
和pytorch
都重装了一遍,能运行就不说了,不能运行的话,可以参考我这篇文章:OSError: [WinError 127] 找不到指定的程序、“caffe2_detectron_ops.dll“ or one of its dependencies.
5 待更新
文章出处登录后可见!
已经登录?立即刷新