赞
踩
注意:这个解决方法是你在尝试了网上各种方法后都没法解决后可能去试一试的方法
简单质朴而通透
我跑的代码推荐torch版本是1.9.0 ,然而这个上古机器的CUDA却是10.2
所以如果你走投无路了,就去查看一下的你的CUDA版本是不是太老以至于配不上你的torch版本
import torch
#import pytorch_lightning as pl
if torch.cuda.is_available():
print("CUDA is available")
else:
print("CUDA is not available")
print(torch.version.cuda)
print(torch.__version__)
输出:
CUDA is available
10.2
1.9.0+cu102
如果对不上,就去更新的CUDA吧(如何更新CUDA教程一大堆)
比如这个更新CUDA的教程
友情附上pytorch的官网,去这里用他的命令行下载torch,要注意区分GPU和CPU
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。