window下的cuda安装

我自己搞了很久的cuda总是报错false 或者显示错误为pytorch module ‘torch’ has no attribute ‘cuda’
现在来记录下 省的下次忘掉
1 首先版本要和自己电脑的driver version对应 可以小不可以超过自己电脑版本
2 检测cuda是否可以使用
import torch print('CUDA版本:',torch.version.cuda) print('Pytorch版本:',torch.__version__) print('显卡是否可用:','可用' if(torch.cuda.is_available()) else '不可用') print('显卡数量:',torch.cuda.device_count()) print('是否支持BF16数字格式:','支持' if (torch.cuda.is_bf16_supported()) else '不支持') print('当前显卡型号:',torch.cuda.get_device_name()) print('当前显卡的CUDA算力:',torch.cuda.get_device_capability()) print('当前显卡的总显存:',torch.cuda.get_device_properties(0).total_memory/1024/1024/1024,'GB') print('是否支持TensorCore:','支持' if (torch.cuda.get_device_properties(0).major >= 7) else '不支持') print('当前显卡的显存使用率:',torch.cuda.memory_allocated(0)/torch.cuda.get_device_properties(0).total_memory*100,'%')
3 报错的话先尝试去除之前的pytorch
conda uninstall pytorch
4 使用pytorch官网的命令进行下载cuda
我的版本是11.8
因此在conda下命令为下面的command （这个是对应命令的官网）

5 最后在用步骤2测试下