先单独创建一个py文件用于测试cuda是否配置正确
print(torch.cuda.is_available())
运行后输出的是True,这代表你的cuda已经安装好了。
打开报错的程序,在主程序里面加入上面那一句:
print(torch.cuda.is_available())
然后把device改成cuda,默认是cuda0
然后运行问题就解决了。
先单独创建一个py文件用于测试cuda是否配置正确
print(torch.cuda.is_available())
运行后输出的是True,这代表你的cuda已经安装好了。
打开报错的程序,在主程序里面加入上面那一句:
print(torch.cuda.is_available())
然后把device改成cuda,默认是cuda0
然后运行问题就解决了。