Skip to content
Discussion options

You must be logged in to vote

另外,nvidia-smi 查询有个缺陷,就是查询的 cuda 版本是系统能支持的最高版本,并非当前版本,当前版本请通过 cat /usr/local/cuda/version.txt 查询,以及驱动使用 cat /proc/driver/nvidia/version 查询。

结束后请检查是否能正常调用 cuda 核心。

import torch
torch.cuda.is_available()
torch.zeros(1).cuda()
# tensor([0.], device='cuda:0') means that cuda kernel is available.

我的本地部署的环境 ubuntu 22.04 GTX1650 版本 CUDA Version 11.0 驱动 470.256.02 。本项目的 torch version 是 1.11.0,请寻找你合适的 cuda version 以及 driver

Replies: 15 comments

Comment options

You must be logged in to vote
0 replies
Comment options

You must be logged in to vote
0 replies
Comment options

You must be logged in to vote
0 replies
Answer selected by timerring
Comment options

You must be logged in to vote
0 replies
Comment options

You must be logged in to vote
0 replies
Comment options

You must be logged in to vote
0 replies
Comment options

You must be logged in to vote
0 replies
Comment options

You must be logged in to vote
0 replies
Comment options

You must be logged in to vote
0 replies
Comment options

You must be logged in to vote
0 replies
Comment options

You must be logged in to vote
0 replies
Comment options

You must be logged in to vote
0 replies
Comment options

You must be logged in to vote
0 replies
Comment options

You must be logged in to vote
0 replies
Comment options

You must be logged in to vote
0 replies
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Category
Q&A
Labels
None yet
2 participants
Converted from issue

This discussion was converted from issue #175 on May 01, 2025 05:42.