Linux——查看cuda版本的全面方法

1. nvcc -V(最准确:查看运行时的 CUDA 版本)

这是最常用且最准确的方法。它显示的是 CUDA Runtime API 的版本,也就是你实际编译代码(例如编译自定义的算子或模型时)所调用的 CUDA Toolkit 版本。

bash 复制代码
nvcc --version
# 或者
nvcc -V

注:如果系统提示找不到命令,可能是环境变量未配置。你可以尝试直接使用默认安装路径运行:/usr/local/cuda/bin/nvcc -V

2. 查看本地版本文件(Linux 常用)

如果服务器上安装了多个 CUDA 版本,可以通过查看安装目录下的版本文件来确认当前默认环境的版本:

bash 复制代码
cat /usr/local/cuda/version.txt
# 对于较新的 CUDA 版本,可能是 json 文件:
cat /usr/local/cuda/version.json

在 Linux 服务器中,/usr/local/cuda 通常是一个软链接,指向实际激活的版本(如 /usr/local/cuda-12.1)。你可以通过 ls -l /usr/local | grep cuda 查看服务器上具体保留了哪些历史版本。

3. 通过代码运行环境查看(PyTorch 环境)

在跑实验前,直接在代码中确认深度学习框架当前绑定的 CUDA 版本是一个很好的习惯,这样可以确保你的追踪框架或多模态大模型是在正确的硬件加速环境下运行的:

python 复制代码
import torch

print(f"PyTorch Version: {torch.__version__}")
print(f"CUDA Version used by PyTorch: {torch.version.cuda}")
print(f"Is CUDA available?: {torch.cuda.is_available()}")

# 这行代码可以顺便确认当前挂载的显卡型号
if torch.cuda.is_available():
    print(f"Current Device: {torch.cuda.get_device_name(0)}") 
相关推荐
ltl9 小时前
内核追踪:ftrace、kprobe、uprobe、tracepoint 生产实战
linux
·薯条大王9 小时前
经济实惠玩云服务器|一台云服务器多人共用,子账号配置教程
java·linux·运维·服务器·汇编·c++·python
ZJH__GO12 小时前
网络编程v4pro--实现聊天室文件传输功能
java·服务器·网络·计算机网络
小小、码农12 小时前
Linux进程控制:从fork到exec,手写一个简易Shell
linux·运维·服务器
不会就选b13 小时前
Linux之进程间通信(二)---模拟进程池
linux·运维·服务器
NeilYuen14 小时前
用UDP实现向DNS服务器请求IP地址
服务器·tcp/ip·udp
沫璃染墨15 小时前
《从零入门Linux系统篇(十五):系统工具篇·六——GDB调试器详解:从程序执行控制到高级调试技巧》
linux·运维·服务器·c语言·c++
AIgorithmGEEK15 小时前
Linux 环境变量完全指南:从入门到理解底层原理
linux·运维·服务器
修罗王16 小时前
Linux Docker 服务管理与排查实战指南(新手速查版)
运维·docker
迪康Defender16 小时前
从静态存储到动态流转:终端透明加密两种模式实战解析
运维·服务器·网络·数据库·其他