使用nvidia-smi命令查看CUDA版本为11.4,nvcc -V命令查看CUDA版本为11.1。以nvcc -V版本为准。
CUDA有两个CUDA版本
runtime api 使用nvcc -V显示 ,是运行版
driver api, 使用nvidia-smi显示,是驱动版
如果报错的命令是RuntimeError,那就使用nvcc -V命令查看是否是版本不匹配。
nvidia-smi 显示的版本是驱动的版本,是原先在容器里已经装好的。
而nvcc -V 显示的版本是cuda toolkit的版本,是需要我们去安装的。
由于驱动版本是向下兼容的, 因此cuda toolkit版本小于等于驱动版本都可以正常使用。
pytorch, cudadnn等要求的cuda版本都是指cuda toolkit的版本。
参考链接:
【CUDA】nvcc和nvidia-smi显示的版本不一致?_nvidia-smi命令列出的cuda版本与nvcc -v列出的版本号不一致-CSDN博客