麻烦老师看看,如何装CUDA驱动,才能调用GPU? #490

Closed
opened 2025-01-23 13:42:01 +08:00 by 12754933808cs · 2 comments

我的CUDA驱动版本是12.4,pytorch版本是2.5.1,都已经手动装好了,接下来要如何做,才能调用GPU?我家有4090的显卡,这样正好可以试。这是我下载的驱动,但依然是GPU用不了

我的CUDA驱动版本是12.4,pytorch版本是2.5.1,都已经手动装好了,接下来要如何做,才能调用GPU?我家有4090的显卡,这样正好可以试。这是我下载的驱动,但依然是GPU用不了
Author

CUDA available: False
No CUDA device found. Using CPU.显示这个

CUDA available: False No CUDA device found. Using CPU.显示这个
  1. 检查 PyTorch 和 CUDA 的兼容性:可以参考 https://pytorch.org/get-started/previous-versions/
  2. 验证 CUDA 是否正确安装: nvcc --version
  3. 验证 PyTorch 是否正确安装:
    import torch
    print(torch.__version__) # 检查 PyTorch 版本
    print(torch.cuda.is_available()) # 检查 CUDA 是否可用
    print(torch.cuda.get_device_name(0)) # 检查 GPU 设备名称
  4. 检查 GPU 驱动:nvidia-smi
    可以参考:https://blog.csdn.net/wokaowokaowokao12345/article/details/131134993
1. 检查 PyTorch 和 CUDA 的兼容性:可以参考 https://pytorch.org/get-started/previous-versions/ 2. 验证 CUDA 是否正确安装: nvcc --version 3. 验证 PyTorch 是否正确安装: `import torch` `print(torch.__version__) # 检查 PyTorch 版本` `print(torch.cuda.is_available()) # 检查 CUDA 是否可用` `print(torch.cuda.get_device_name(0)) # 检查 GPU 设备名称` 4. 检查 GPU 驱动:nvidia-smi 可以参考:https://blog.csdn.net/wokaowokaowokao12345/article/details/131134993
Sign in to join this conversation.
No Milestone
No project
No Assignees
2 Participants
Notifications
Due Date
The due date is invalid or out of range. Please use the format 'yyyy-mm-dd'.

No due date set.

Dependencies

No dependencies set.

Reference: HswOAuth/llm_course#490
No description provided.