问题:下载timm,将torch升级版本导致无法使用gpu 

import torch
print(torch.cuda.is_available())

代码返回false。

此外,使用conda命令安装timm找不到包,使用pip安装会自动更新torch

(让我头疼很久不知道怎么办,网上也没个解决办法,不断尝试了各个版本,关键是版本好些还瞎下载不下来 晕(((φ(◎ロ◎;)φ))))

解决方法:安装顺序问题,先安装timm,再安装torch

分析原因:timm依赖于torch,但是torch不依赖timm,先安装timm后再安装torch可以自动调整版本

关于torch.cuda.is_available()返回false,引用一则博客(大家对应自取,可能porch版本安装错误):torch.cuda.is_available()返回False可能不是因为cuda或者cudnn没装好_没装cuda为什么cuda available-CSDN博客

Logo

有“AI”的1024 = 2048,欢迎大家加入2048 AI社区

更多推荐