weixin_46492009的博客

原创 有的虚拟环境cuda返回true有的返回false的解决办法

print(torch.__version__)检查torch版本，发现print(torch.cuda.is_available())返回false的torch版本是1.13.1+cpu，原来”pip install torch -i 清华镜像“这样安装的是cpu版本的torch。而print(torch.cuda.is_available())返回true的torch版本是1.13.1。安装对应cuda版本的torch和torchvision，我的cuda是11.7。