部署大模型,验证cuda是否可用以及版本
发表日期:2025-02-23 06:04:19 | 来源: | | 浏览(712) 分类:AI大模型
部署大模型的时候经常因为cuda原因不能运行,所以我们最好先查验一下自己的环境。
新建Python文件,start.py 然后运行一下
import torch
print(f"CUDA 是否可用: {torch.cuda.is_available()}")
print(f"PyTorch 版本: {torch.__version__}")
print(f"检测到的 CUDA 版本: {torch.version.cuda}")
cmd 里输入:nvidia-smi 查看一下自己的显卡驱动版本和cuda支持版本

宁公网安备 64010402001209号