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())"