部署大模型,验证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}")

image.pngcmd 里输入:nvidia-smi 查看一下自己的显卡驱动版本和cuda支持版本

image.png


集速网 copyRight © 2015-2022 宁ICP备15000399号-1 宁公网安备 64010402001209号
与其临渊羡鱼,不如退而结网
欢迎转载、分享、引用、推荐、收藏。