已安装 CUDA 与Pytorch 但 torch.cuda.is_available() 为 False

netkiller-BG7NYT

已于 2025-03-17 16:34:00 修改

阅读量1.6k

点赞数 4

CC 4.0 BY-SA版权

分类专栏： Python 手札文章标签： pytorch 人工智能 python

于 2024-11-17 14:25:20 首次发布

本文链接：https://blog.csdn.net/u010604770/article/details/143831649

Python 手札专栏收录该内容

65 篇文章 ¥9.90 ¥99.00

订阅专栏

超级会员免费看

已 CUDA 与Pytorch 但 torch.cuda.is_available() 为 False

检查 CUDA 是否可用

device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
print(device)	

输出结果

D:\workspace\netkiller\.venv\Scripts\python.exe D:\workspace\netkiller\test\duda.py 
cpu

表示当前使用的的是cpu,而不是 GPU

硬件状态检测

使用 nvidia-smi 命令检测显卡，是否支持 CUDA

PS C:\Users\neo> nvidia-smi
Sun Nov 17 12:29:15 2024
+---------------------------------------------------------------------------------------+
| NVIDIA-SMI 546.67                 Driver Version: 546.67       CUDA Version: 12.3     |
|-----------------------------------------+----------------------+----------------------+
| GPU  Name                     TCC/WDDM  | Bus-I

了解本专栏