(A3)测试cuda是否可用

1、测试cuda是否可用

python -c "import torch; torch.cuda.is_available(); torch.zeros(1).cuda()"

说明
cuda支持的算力上限可以低于显卡的算力上限,但是差距不要太大。
之前用torch0.4.1+cuda9.2+显卡3090,发现cuda不可用,从网上找到的资料分析是算力不匹配。
3090显卡使用

2、测试cuda支持的算力上限

在 torch 版本较高时可以使用

python -c "import torch; print(torch.cuda.get_arch_list())"

你可能感兴趣的:(常用指令积累,深度学习,pytorch,python)