部署大模型,验证cuda是否可用以及版本
发表日期:2025-02-23 06:04:19 | 来源: | | 浏览(38) 分类: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支持版本