安装GPU版本Pytorch失败:torch.cuda.is_available()为False
问题所在
检查conda list发现,实际安装的Pytorch为CPU版本(虽然安装时明确指定了cuda版本):
上图中可以看出,Pytorch的描述为:py3.9_cpu_0
解决办法
有可能是因为环境中存在一个叫“cpuonly”的包,导致无法安装GPU版本Pytorch:
卸载掉它即可,卸载它时会自动将Pytorch更新为GPU版本:
如果检查发现环境中其实并没有这个包呢?可以借鉴此方法,先安装一个“cpuonly”包,再卸载掉它,Pytorch也会自动更新为GPU版本。
conda unistall pytorch conda istall pytorch
本文作者:yutian
本文链接:https://www.cnblogs.com/yutian-blogs/p/17971315
版权声明:本作品采用知识共享署名-非商业性使用-禁止演绎 2.5 中国大陆许可协议进行许可。
【推荐】国内首个AI IDE,深度理解中文开发场景,立即下载体验Trae
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步